fix

EnzymeAD · Dec 14, 2024 · 939ddc0 · 939ddc0
1 parent 2d3f43b
commit 939ddc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Enzyme.jl b/src/Enzyme.jl
@@ -687,7 +687,7 @@ code, as well as high-order differentiation.
     A2 = A
 
     if A isa UnionAll
-        rt = Compiler.primal_return_type(mode, FTy, tt)
+        rt = Compiler.primal_return_type(Reverse, FTy, tt)
         A2 = A{rt}
         if rt == Union{}
             rt = Nothing