break if replacement found

Author: aidint

mlir/lib/Reducer/ReductionTreePass.cpp

@@ -68,8 +68,10 @@ static void applyPatterns(Region &region,
         size_t index = operandTie.index();
         auto operand = operandTie.value();
         for (auto candidate : valueMap[operand.getType()])
-          if (domInfo.properlyDominates(candidate, op))
+          if (domInfo.properlyDominates(candidate, op)) {
             op->setOperand(index, candidate);
+            break;
+          }
       }
 
     // `applyOpPatternsGreedily` with folding returns whether the op is

mlir/test/mlir-reduce/replace-operands.mlir

@@ -3,7 +3,7 @@
 
 // CHECK-LABEL: func.func @main
 func.func @main() {
-  // CHECK-NEXT: %[[RESULT:.*]] = arith.constant 2 : i32
+  // CHECK-NEXT: %[[RESULT:.*]] = arith.constant 3 : i32
   // CHECK-NEXT: {{.*}} = "test.op_crash"(%[[RESULT]], %[[RESULT]]) : (i32, i32) -> i32
   // CHECK-NEXT return