Update customrules.jl

EnzymeAD · Dec 6, 2024 · 38fcba4 · 38fcba4
1 parent df785e0
commit 38fcba4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rules/customrules.jl b/src/rules/customrules.jl
@@ -111,9 +111,9 @@ function enzyme_custom_setup_args(
         if cmode == API.DEM_ReverseModeGradient
           cmode = API.DEM_ReverseModePrimal
         end
+        uncacheable = Vector{UInt8}(undef, length(collect(LLVM.operands(op))) - 1)
 	activep =
 		if mode == API.DEM_ForwardMode ||
-    		   uncacheable = Vector{UInt8}(undef, length(collect(LLVM.operands(op))) - 1)
 		   API.EnzymeGradientUtilsGetUncacheableArgs(
 		    gutils,
 		    op,