Improve logs and use workaround for runtime activity

TuringLang · Aug 21, 2023 · 6da01ea · 6da01ea
1 parent 5fb9422
commit 6da01ea
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test/ad/utils.jl b/test/ad/utils.jl
@@ -6,7 +6,13 @@ const FDM = FiniteDifferences
 
 # Load AD backends
 if GROUP == "All" || GROUP == "Enzyme"
-    @eval using Enzyme
+    @eval begin
+        using Enzyme
+        # Disable Enzyme warnings
+        Enzyme.API.typeWarning!(false)
+        # Enable runtime activity (workaround)
+        Enzyme.API.runtimeActivity!(true)
+    end
 end
 if GROUP == "All" || GROUP == "ForwardDiff"
     @eval using ForwardDiff