From e6d146c7be322e33c6cd314dcb447d0bba972ebb Mon Sep 17 00:00:00 2001
From: "William S. Moses" <gh@wsmoses.com>
Date: Tue, 8 Oct 2024 22:46:01 -0400
Subject: [PATCH] fix

---
 enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.cpp | 6 +++---
 enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.h   | 3 +--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.cpp b/enzyme/Enzyme/MLIR/Interfaces/GradientUtilsReverse.cpp
index a0add5dcff57..6db56dc52fdf 100644
--- 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
index bf5b21588b5b..cb54a3673281 100644
--- 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);
 };