fix: return types for Enzyme ForwardDiff

EnzymeAD · Dec 19, 2024 · edf2d71 · edf2d71
1 parent 1c9e3e8
commit edf2d71
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/src/Interpreter.jl b/src/Interpreter.jl
@@ -283,7 +283,7 @@ function overload_autodiff(
         end
     end
     for (i, act) in enumerate(activity)
-        if act == enzyme_out || (reverse && (act == enzyme_dup || act == enzyme_dupnoneed))
+        if act == enzyme_out || act == enzyme_dup || act == enzyme_dupnoneed
             if width == 1
                 push!(outtys, in_tys[i])
             else

diff --git a/test/autodiff.jl b/test/autodiff.jl
@@ -11,7 +11,7 @@ fwd(Mode, RT, x, y) = Enzyme.autodiff(Mode, square, RT, Duplicated(x, y))
 
     res1 = @jit(
         fwd(
-            set_abi(Forward, Reactant.ReactantABI),
+            Forward,
             Duplicated,
             ConcreteRArray(ones(3, 2)),
             ConcreteRArray(3.1 * ones(3, 2)),
@@ -42,12 +42,7 @@ fwd(Mode, RT, x, y) = Enzyme.autodiff(Mode, square, RT, Duplicated(x, y))
     @test typeof(ores1) == Tuple{}
 
     res1 = @jit(
-        fwd(
-            set_abi(Forward, Reactant.ReactantABI),
-            Const,
-            ConcreteRArray(ones(3, 2)),
-            ConcreteRArray(3.1 * ones(3, 2)),
-        )
+        fwd(Forward, Const, ConcreteRArray(ones(3, 2)), ConcreteRArray(3.1 * ones(3, 2)))
     )
 
     @test typeof(res1) == Tuple{}