From 348303f696a729e8a7dc44fe7784add05fc83a36 Mon Sep 17 00:00:00 2001
From: "William S. Moses" <gh@wsmoses.com>
Date: Sat, 30 Nov 2024 19:51:48 -0500
Subject: [PATCH] fixup

---
 src/rules/customrules.jl | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/src/rules/customrules.jl b/src/rules/customrules.jl
index f22c9842a8..8a56481a82 100644
--- a/src/rules/customrules.jl
+++ b/src/rules/customrules.jl
@@ -923,15 +923,10 @@ function enzyme_custom_common_rev(
     @assert ami !== nothing
     target = DefaultCompilerTarget()
     params = PrimalCompilerParams(mode)
-    aug_RT = something(
-        Core.Compiler.typeinf_type(
-            GPUCompiler.get_interpreter(
-                CompilerJob(ami, CompilerConfig(target, params; kernel = false), world),
-            ),
-            ami,
-        ),
-        Any,
+    interp = GPUCompiler.get_interpreter(
+        CompilerJob(ami, CompilerConfig(target, params; kernel = false), world),
     )
+    aug_RT = return_type(interp, ami)
     if kwtup !== nothing && kwtup <: Duplicated
         @safe_debug "Non-constant keyword argument found for " augprimal_TT
         emit_error(