fix: forward enzyme rules

LuxDL · Sep 22, 2024 · 8459414 · 8459414
1 parent e977188
commit 8459414
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/utils.jl b/src/utils.jl
@@ -252,8 +252,7 @@ macro enzyme_alternative(f₁, f₂)
 
         function EnzymeRules.forward(cfg::EnzymeRules.FwdConfig,
                 ::EnzymeCore.Const{typeof($(f₁))}, ::Type{RT}, args...) where {RT}
-            EnzymeCore.autodiff(cfg, EnzymeCore.Forward, EnzymeCore.Const($(f₂)), RT,
-                args...)
+            EnzymeCore.autodiff(EnzymeCore.Forward, EnzymeCore.Const($(f₂)), RT, args...)
             return
         end
     end)