Skip to content

Commit 634c57d

Browse files
[mlir][SCF][NFC] scf.for/scf.while: rename builder args (#111493)
Rename builder args to make them consistent with the `args` in the TableGen definition.
1 parent 20b8d3f commit 634c57d

File tree

2 files changed

+13
-13
lines changed

2 files changed

+13
-13
lines changed

mlir/include/mlir/Dialect/SCF/IR/SCFOps.td

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ def ForOp : SCF_Op<"for",
249249
let skipDefaultBuilders = 1;
250250
let builders = [
251251
OpBuilder<(ins "Value":$lowerBound, "Value":$upperBound, "Value":$step,
252-
CArg<"ValueRange", "std::nullopt">:$iterArgs,
252+
CArg<"ValueRange", "std::nullopt">:$initArgs,
253253
CArg<"function_ref<void(OpBuilder &, Location, Value, ValueRange)>",
254254
"nullptr">)>
255255
];
@@ -1074,7 +1074,7 @@ def WhileOp : SCF_Op<"while",
10741074
let regions = (region SizedRegion<1>:$before, SizedRegion<1>:$after);
10751075

10761076
let builders = [
1077-
OpBuilder<(ins "TypeRange":$resultTypes, "ValueRange":$operands,
1077+
OpBuilder<(ins "TypeRange":$resultTypes, "ValueRange":$inits,
10781078
"function_ref<void(OpBuilder &, Location, ValueRange)>":$beforeBuilder,
10791079
"function_ref<void(OpBuilder &, Location, ValueRange)>":$afterBuilder)>
10801080
];

mlir/lib/Dialect/SCF/IR/SCF.cpp

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -312,25 +312,25 @@ void ConditionOp::getSuccessorRegions(
312312
//===----------------------------------------------------------------------===//
313313

314314
void ForOp::build(OpBuilder &builder, OperationState &result, Value lb,
315-
Value ub, Value step, ValueRange iterArgs,
315+
Value ub, Value step, ValueRange initArgs,
316316
BodyBuilderFn bodyBuilder) {
317317
OpBuilder::InsertionGuard guard(builder);
318318

319319
result.addOperands({lb, ub, step});
320-
result.addOperands(iterArgs);
321-
for (Value v : iterArgs)
320+
result.addOperands(initArgs);
321+
for (Value v : initArgs)
322322
result.addTypes(v.getType());
323323
Type t = lb.getType();
324324
Region *bodyRegion = result.addRegion();
325325
Block *bodyBlock = builder.createBlock(bodyRegion);
326326
bodyBlock->addArgument(t, result.location);
327-
for (Value v : iterArgs)
327+
for (Value v : initArgs)
328328
bodyBlock->addArgument(v.getType(), v.getLoc());
329329

330330
// Create the default terminator if the builder is not provided and if the
331331
// iteration arguments are not provided. Otherwise, leave this to the caller
332332
// because we don't know which values to return from the loop.
333-
if (iterArgs.empty() && !bodyBuilder) {
333+
if (initArgs.empty() && !bodyBuilder) {
334334
ForOp::ensureTerminator(*bodyRegion, builder, result.location);
335335
} else if (bodyBuilder) {
336336
OpBuilder::InsertionGuard guard(builder);
@@ -3260,23 +3260,23 @@ LogicalResult ReduceReturnOp::verify() {
32603260

32613261
void WhileOp::build(::mlir::OpBuilder &odsBuilder,
32623262
::mlir::OperationState &odsState, TypeRange resultTypes,
3263-
ValueRange operands, BodyBuilderFn beforeBuilder,
3263+
ValueRange inits, BodyBuilderFn beforeBuilder,
32643264
BodyBuilderFn afterBuilder) {
3265-
odsState.addOperands(operands);
3265+
odsState.addOperands(inits);
32663266
odsState.addTypes(resultTypes);
32673267

32683268
OpBuilder::InsertionGuard guard(odsBuilder);
32693269

32703270
// Build before region.
32713271
SmallVector<Location, 4> beforeArgLocs;
3272-
beforeArgLocs.reserve(operands.size());
3273-
for (Value operand : operands) {
3272+
beforeArgLocs.reserve(inits.size());
3273+
for (Value operand : inits) {
32743274
beforeArgLocs.push_back(operand.getLoc());
32753275
}
32763276

32773277
Region *beforeRegion = odsState.addRegion();
3278-
Block *beforeBlock = odsBuilder.createBlock(
3279-
beforeRegion, /*insertPt=*/{}, operands.getTypes(), beforeArgLocs);
3278+
Block *beforeBlock = odsBuilder.createBlock(beforeRegion, /*insertPt=*/{},
3279+
inits.getTypes(), beforeArgLocs);
32803280
if (beforeBuilder)
32813281
beforeBuilder(odsBuilder, odsState.location, beforeBlock->getArguments());
32823282

0 commit comments

Comments
 (0)