diff --git a/mlir/include/mlir/Dialect/SCF/SCFOps.td b/mlir/include/mlir/Dialect/SCF/SCFOps.td index 1245102d4c3c..28348f083f16 100644 --- a/mlir/include/mlir/Dialect/SCF/SCFOps.td +++ b/mlir/include/mlir/Dialect/SCF/SCFOps.td @@ -171,6 +171,10 @@ def ForOp : SCF_Op<"for", Operation::operand_range getIterOperands() { return getOperands().drop_front(getNumControlOperands()); } + MutableArrayRef getIterOpOperands() { + return + getOperation()->getOpOperands().drop_front(getNumControlOperands()); + } void setLowerBound(Value bound) { getOperation()->setOperand(0, bound); } void setUpperBound(Value bound) { getOperation()->setOperand(1, bound); }