llvm::SmallVector --> ArrayRef

EnzymeAD · Dec 22, 2024 · aa4e598 · aa4e598
1 parent 8a3f3c7
commit aa4e598
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/enzyme/Enzyme/MLIR/Dialect/EnzymeOps.td b/enzyme/Enzyme/MLIR/Dialect/EnzymeOps.td
@@ -204,7 +204,7 @@ def BroadcastOp : Enzyme_Op<"broadcast"> {
   let results = (outs AnyRankedTensor:$output);
 
   let builders = [
-    OpBuilder<(ins "Value":$input, "llvm::SmallVector<int64_t>":$shape)>
+    OpBuilder<(ins "Value":$input, "ArrayRef<int64_t>":$shape)>
   ];
 }
 

diff --git a/enzyme/Enzyme/MLIR/Dialect/Ops.cpp b/enzyme/Enzyme/MLIR/Dialect/Ops.cpp
@@ -196,7 +196,7 @@ LogicalResult BatchOp::verifySymbolUses(SymbolTableCollection &symbolTable) {
 // BroadcastOp
 //===----------------------------------------------------------------------===//
 
-void BroadcastOp::build(OpBuilder &builder, OperationState &result, Value input, llvm::SmallVector<int64_t> shape) {
+void BroadcastOp::build(OpBuilder &builder, OperationState &result, Value input, ArrayRef<int64_t> shape) {
   auto shapeAttr = builder.getDenseI64ArrayAttr(shape);
   RankedTensorType output;
   // TODO: support things other than scalars and ranked tensors, maybe reuse getShadowType here?