From 2d3f43ba696e87d604dfdb571517c3c065313bd1 Mon Sep 17 00:00:00 2001
From: "William S. Moses" <gh@wsmoses.com>
Date: Sat, 14 Dec 2024 16:58:14 -0500
Subject: [PATCH] fixup

---
 src/Enzyme.jl | 2 +-
 src/utils.jl  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Enzyme.jl b/src/Enzyme.jl
index 4328a561e3..e2244171a5 100644
--- 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
index 4802d2acc6..aa94e766f8 100644
--- 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