From 339316fa4889ff5010f7c146667e9ba4f308e7a7 Mon Sep 17 00:00:00 2001 From: "William S. Moses" Date: Sat, 14 Dec 2024 17:39:19 -0500 Subject: [PATCH] fix --- src/compiler.jl | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/compiler.jl b/src/compiler.jl index c13f8e7cad..2d9b12630c 100644 --- a/src/compiler.jl +++ b/src/compiler.jl @@ -3556,13 +3556,13 @@ function GPUCompiler.codegen( caller = mi if mode == API.DEM_ForwardMode has_custom_rule = - EnzymeRules.has_frule_from_sig(specTypes; world, method_table, caller) + EnzymeRules.has_frule_from_sig(specTypes; world, method_table) # , caller) if has_custom_rule @safe_debug "Found frule for" mi.specTypes end else has_custom_rule = - EnzymeRules.has_rrule_from_sig(specTypes; world, method_table, caller) + EnzymeRules.has_rrule_from_sig(specTypes; world, method_table) # , caller) if has_custom_rule @safe_debug "Found rrule for" mi.specTypes end @@ -3577,7 +3577,7 @@ function GPUCompiler.codegen( actualRetType = k.ci.rettype end - if EnzymeRules.noalias_from_sig(mi.specTypes; world, method_table, caller) + if EnzymeRules.noalias_from_sig(mi.specTypes; world, method_table) #, caller) push!(return_attributes(llvmfn), EnumAttribute("noalias")) for u in LLVM.uses(llvmfn) c = LLVM.user(u) @@ -3801,7 +3801,7 @@ end end continue end - if EnzymeRules.is_inactive_from_sig(specTypes; world, method_table, caller) && + if EnzymeRules.is_inactive_from_sig(specTypes; world, method_table) && # , caller) && Enzyme.has_method( Tuple{typeof(EnzymeRules.inactive),specTypes.parameters...}, world, @@ -3819,7 +3819,7 @@ end ) continue end - if EnzymeRules.is_inactive_noinl_from_sig(specTypes; world, method_table, caller) && + if EnzymeRules.is_inactive_noinl_from_sig(specTypes; world, method_table) && #, caller) && has_method( Tuple{typeof(EnzymeRules.inactive_noinl),specTypes.parameters...}, world,