fix

impICNF · Oct 3, 2024 · ffe1684 · ffe1684
1 parent c5a04a6
commit ffe1684
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 14 deletions.
diff --git a/README.md b/README.md
@@ -50,7 +50,7 @@ icnf = construct(
     nn,
     nvars, # number of variables
     naugs; # number of augmented dimensions
-    # compute_mode = DIJacVecMatrixMode(AutoEnzyme(; mode = set_runtime_activity(Enzyme.Forward), function_annotation = Enzyme.Const)), # process data in batches and use Enzyme
+    # compute_mode = DIJacVecMatrixMode(AutoEnzyme(; mode = Enzyme.set_runtime_activity(Enzyme.Forward), function_annotation = Enzyme.Const)), # process data in batches and use Enzyme
     # inplace = true, # use the inplace version of functions
     # resource = CUDALibs(), # process data by GPU
     tspan = (0.0f0, 13.0f0), # have bigger time span

diff --git a/benchmark/benchmarks.jl b/benchmark/benchmarks.jl
@@ -36,7 +36,7 @@ icnf = ContinuousNormalizingFlows.construct(
     naugs;
     compute_mode = ContinuousNormalizingFlows.DIJacVecMatrixMode(
         ADTypes.AutoEnzyme(;
-            mode = set_runtime_activity(Enzyme.Forward),
+            mode = Enzyme.set_runtime_activity(Enzyme.Forward),
             function_annotation = Enzyme.Const,
         ),
     ),
@@ -87,7 +87,7 @@ icnf2 = ContinuousNormalizingFlows.construct(
     inplace = true,
     compute_mode = ContinuousNormalizingFlows.DIJacVecMatrixMode(
         ADTypes.AutoEnzyme(;
-            mode = set_runtime_activity(Enzyme.Forward),
+            mode = Enzyme.set_runtime_activity(Enzyme.Forward),
             function_annotation = Enzyme.Const,
         ),
     ),

diff --git a/src/base_icnf.jl b/src/base_icnf.jl
@@ -6,7 +6,7 @@ function construct(
     data_type::Type{<:AbstractFloat} = Float32,
     compute_mode::ComputeMode = DIJacVecMatrixMode(
         ADTypes.AutoEnzyme(;
-            mode = set_runtime_activity(Enzyme.Forward),
+            mode = Enzyme.set_runtime_activity(Enzyme.Forward),
             function_annotation = Enzyme.Const,
         ),
     ),

diff --git a/test/call_tests.jl b/test/call_tests.jl
@@ -37,25 +37,25 @@ Test.@testset "Call Tests" begin
         ContinuousNormalizingFlows.DIJacVecMatrixMode(ADTypes.AutoZygote()),
         ContinuousNormalizingFlows.DIVecJacVectorMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Reverse),
+                mode = Enzyme.set_runtime_activity(Enzyme.Reverse),
                 function_annotation = Enzyme.Const,
             ),
         ),
         ContinuousNormalizingFlows.DIJacVecVectorMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Forward),
+                mode = Enzyme.set_runtime_activity(Enzyme.Forward),
                 function_annotation = Enzyme.Const,
             ),
         ),
         ContinuousNormalizingFlows.DIVecJacMatrixMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Reverse),
+                mode = Enzyme.set_runtime_activity(Enzyme.Reverse),
                 function_annotation = Enzyme.Const,
             ),
         ),
         ContinuousNormalizingFlows.DIJacVecMatrixMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Forward),
+                mode = Enzyme.set_runtime_activity(Enzyme.Forward),
                 function_annotation = Enzyme.Const,
             ),
         ),

diff --git a/test/fit_tests.jl b/test/fit_tests.jl
@@ -34,25 +34,25 @@ Test.@testset "Fit Tests" begin
         ContinuousNormalizingFlows.DIJacVecMatrixMode(ADTypes.AutoZygote()),
         ContinuousNormalizingFlows.DIVecJacVectorMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Reverse),
+                mode = Enzyme.set_runtime_activity(Enzyme.Reverse),
                 function_annotation = Enzyme.Const,
             ),
         ),
         ContinuousNormalizingFlows.DIJacVecVectorMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Forward),
+                mode = Enzyme.set_runtime_activity(Enzyme.Forward),
                 function_annotation = Enzyme.Const,
             ),
         ),
         ContinuousNormalizingFlows.DIVecJacMatrixMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Reverse),
+                mode = Enzyme.set_runtime_activity(Enzyme.Reverse),
                 function_annotation = Enzyme.Const,
             ),
         ),
         ContinuousNormalizingFlows.DIJacVecMatrixMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Forward),
+                mode = Enzyme.set_runtime_activity(Enzyme.Forward),
                 function_annotation = Enzyme.Const,
             ),
         ),

diff --git a/test/instability_tests.jl b/test/instability_tests.jl
@@ -18,7 +18,7 @@ Test.@testset "Instability" begin
         naugs;
         compute_mode = ContinuousNormalizingFlows.DIJacVecMatrixMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Forward),
+                mode = Enzyme.set_runtime_activity(Enzyme.Forward),
                 function_annotation = Enzyme.Const,
             ),
         ),

diff --git a/test/regression_tests.jl b/test/regression_tests.jl
@@ -13,7 +13,7 @@ Test.@testset "Regression Tests" begin
         naugs;
         compute_mode = ContinuousNormalizingFlows.DIJacVecMatrixMode(
             ADTypes.AutoEnzyme(;
-                mode = set_runtime_activity(Enzyme.Forward),
+                mode = Enzyme.set_runtime_activity(Enzyme.Forward),
                 function_annotation = Enzyme.Const,
             ),
         ),