fix

EnzymeAD · Oct 9, 2024 · 15d1ef9 · 15d1ef9
1 parent ed71cab
commit 15d1ef9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.cpp b/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.cpp
@@ -171,8 +171,8 @@ MGradientUtilsReverse *MGradientUtilsReverse::CreateFromClone(
       prefix + todiff.getName(), originalToNew, originalToNewOps,
       additionalArg);
 
-  return new MGradientUtilsReverse(Logic, newFunc, todiff, TA, invertedPointers,
-                                   constant_values, nonconstant_values, retType,
-                                   constant_args, originalToNew,
-                                   originalToNewOps, mode_, width);
+  return new MGradientUtilsReverse(
+      Logic, newFunc, todiff, TA, invertedPointers, returnPrimals,
+      returnShadows, constant_values, nonconstant_values, retType,
+      constant_args, originalToNew, originalToNewOps, mode_, width);
 }