fixup

EnzymeAD · Dec 14, 2024 · 2d3f43b · 2d3f43b
1 parent 9e50cc5
commit 2d3f43b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Enzyme.jl b/src/Enzyme.jl
@@ -370,7 +370,7 @@ Enzyme.autodiff(ReverseWithPrimal, x->x*x, Active(3.0))
     FTy = Core.Typeof(f.val)
 
     rt = if A isa UnionAll
-        Compiler.primal_return_type(mode, FTy, tt)
+        Compiler.primal_return_type(Reverse, FTy, tt)
     else
         eltype(A)
     end

diff --git a/src/utils.jl b/src/utils.jl
@@ -320,7 +320,7 @@ end
 # XXX: version of Base.method_instance that uses a function type
 @inline function my_methodinstance(@nospecialize(mode::Union{Nothing, EnzymeCore.ForwardMode, EnzymeCore.ReverseMode}), @nospecialize(ft::Type), @nospecialize(tt::Type))::Core.MethodInstance
     sig = GPUCompiler.signature_type_by_tt(ft, tt)
-    return prevmethodinstance(mode, ft, tt, world)::Core.MethodInstance
+    return prevmethodinstance(mode, ft, tt)::Core.MethodInstance
 end
 
 export my_methodinstance