fix

EnzymeAD · Oct 9, 2024 · e6d146c · e6d146c
1 parent 2ea5021
commit e6d146c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.cpp b/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.cpp
@@ -134,9 +134,9 @@ void MGradientUtilsReverse::createReverseModeBlocks(Region &oldFunc,
 MGradientUtilsReverse *MGradientUtilsReverse::CreateFromClone(
     MEnzymeLogic &Logic, DerivativeMode mode_, unsigned width,
     FunctionOpInterface todiff, MTypeAnalysis &TA, MFnTypeInfo &oldTypeInfo,
-    const std::vector<bool> &returnPrimals,
-    const std::vector<bool> &returnShadows, ArrayRef<DIFFE_TYPE> retType,
-    ArrayRef<DIFFE_TYPE> constant_args, mlir::Type additionalArg) {
+    const ArrayRef<bool> returnPrimals, const ArrayRef<bool> returnShadows,
+    ArrayRef<DIFFE_TYPE> retType, ArrayRef<DIFFE_TYPE> constant_args,
+    mlir::Type additionalArg) {
   std::string prefix;
 
   switch (mode_) {

diff --git a/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.h b/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.h
@@ -62,8 +62,7 @@ class MGradientUtilsReverse : public MDiffeGradientUtils {
   static MGradientUtilsReverse *CreateFromClone(
       MEnzymeLogic &Logic, DerivativeMode mode_, unsigned width,
       FunctionOpInterface todiff, MTypeAnalysis &TA, MFnTypeInfo &oldTypeInfo,
-      const std::vector<bool> &returnPrimals,
-      const std::vector<bool> &returnShadows,
+      const ArrayRef<bool> returnPrimals, const ArrayRef<bool> returnShadows,
       llvm::ArrayRef<DIFFE_TYPE> retType,
       llvm::ArrayRef<DIFFE_TYPE> constant_args, mlir::Type additionalArg);
 };