Skip to content

Commit c4e1f79

Browse files
committed
address review comments
1 parent 9874741 commit c4e1f79

File tree

5 files changed

+5
-7
lines changed

5 files changed

+5
-7
lines changed

mlir/include/mlir/IR/Builders.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -567,7 +567,7 @@ class OpBuilder : public Builder {
567567
}
568568

569569
/// Attempts to fold the given operation and places new results within
570-
/// 'results'. Returns success if the operation was folded, failure otherwise.
570+
/// `results`. Returns success if the operation was folded, failure otherwise.
571571
/// If the fold was in-place, `results` will not be filled.
572572
/// Note: This function does not erase the operation on a successful fold.
573573
LogicalResult tryFold(Operation *op, SmallVectorImpl<Value> &results);

mlir/lib/IR/Builders.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -478,7 +478,7 @@ Operation *OpBuilder::create(Location loc, StringAttr opName,
478478

479479
LogicalResult OpBuilder::tryFold(Operation *op,
480480
SmallVectorImpl<Value> &results) {
481-
assert(results.empty());
481+
assert(results.empty() && "expected empty results");
482482
ResultRange opResults = op->getResults();
483483

484484
results.reserve(opResults.size());

mlir/test/Transforms/test-legalizer.mlir

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -433,7 +433,7 @@ func.func @use_of_replaced_bbarg(%arg0: i64) {
433433
// CHECK-LABEL: @fold_legalization
434434
func.func @fold_legalization() -> i32 {
435435
// CHECK: op_in_place_self_fold
436-
// CHECK-SAME: folded = true
436+
// CHECK-SAME: folded
437437
%1 = "test.op_in_place_self_fold"() : () -> (i32)
438438
"test.return"(%1) : (i32) -> ()
439439
}

mlir/test/lib/Dialect/Test/TestOps.td

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1352,7 +1352,7 @@ def TestOpInPlaceFold : TEST_Op<"op_in_place_fold"> {
13521352
}
13531353

13541354
def TestOpInPlaceSelfFold : TEST_Op<"op_in_place_self_fold"> {
1355-
let arguments = (ins OptionalAttr<BoolAttr>:$folded);
1355+
let arguments = (ins UnitAttr:$folded);
13561356
let results = (outs I32);
13571357
let hasFolder = 1;
13581358
}

mlir/test/lib/Dialect/Test/TestPatterns.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1170,9 +1170,7 @@ struct TestLegalizePatternDriver
11701170

11711171
// Create a dynamically legal rule that can only be legalized by folding it.
11721172
target.addDynamicallyLegalOp<TestOpInPlaceSelfFold>(
1173-
[](TestOpInPlaceSelfFold op) {
1174-
return op.getProperties().folded != nullptr;
1175-
});
1173+
[](TestOpInPlaceSelfFold op) { return op.getFolded(); });
11761174

11771175
// Handle a partial conversion.
11781176
if (mode == ConversionMode::Partial) {

0 commit comments

Comments
 (0)