attempt to use PrecompileTools

EnzymeAD · Jul 21, 2024 · 0e905d7 · 0e905d7
1 parent 84d8120
commit 0e905d7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/src/Enzyme.jl b/src/Enzyme.jl
@@ -1471,4 +1471,13 @@ macro import_rrule(args...)
     return _import_rrule(args...)
 end
 
+# using PrecompileTools
+# Crashes on 1.11
+# @setup_workload let
+#     @compile_workload begin
+#         autodiff(ReverseMode{false,InlineABI,false}(), ()->nothing, Const)
+#         autodiff(ForwardMode{InlineABI}(), ()->nothing, Const)
+#     end
+# end
+
 end # module
diff --git a/test/precompile.jl b/test/precompile.jl
@@ -42,8 +42,10 @@ precompile_test_harness("Inference caching") do load_path
 
         @setup_workload begin
             @compile_workload begin
-                autodiff(Reverse, mul, Active, Active(1.0), Active(2.0))
-                autodiff(Forward, mul, Duplicated, Duplicated(1.0, 1.0), Const(2.0))
+                autodiff(ReverseMode{false,InlineABI,false}(), mul, Active, Active(1.0), Active(2.0))
+                # Non-Inline mode uses `@generated` functions and poisons the caller
+                # autodiff(Reverse, mul, Active, Active(1.0), Active(2.0))
+                # autodiff(Forward, mul, Duplicated, Duplicated(1.0, 1.0), Const(2.0))
             end
         end
     end) |> string)
@@ -53,7 +55,8 @@ precompile_test_harness("Inference caching") do load_path
         using InferenceCaching
         using Enzyme
 
-        autodiff(Reverse, InferenceCaching.mul, Active, Active(1.0), Active(2.0))
-        autodiff(Forward, InferenceCaching.mul, Duplicated, Duplicated(1.0, 1.0), Const(2.0))
+        @test autodiff(ReverseMode{false,InlineABI,false}(), InferenceCaching.mul, Active, Active(1.0), Active(2.0)) == ((2.0, 1.0),)
+        # autodiff(Reverse, InferenceCaching.mul, Active, Active(1.0), Active(2.0))
+        # autodiff(Forward, InferenceCaching.mul, Duplicated, Duplicated(1.0, 1.0), Const(2.0))
     end
 end