Add scf.while canonicalization patterns to populateUpliftWhileToForPatterns.

Author: NexMing

mlir/lib/Dialect/SCF/Transforms/UpliftWhileToFor.cpp

@@ -399,4 +399,5 @@ FailureOr<scf::ForOp> mlir::scf::upliftWhileToForLoop(RewriterBase &rewriter,
 
 void mlir::scf::populateUpliftWhileToForPatterns(RewritePatternSet &patterns) {
   patterns.add<WhileMoveIfDown, UpliftWhileOp>(patterns.getContext());
+  scf::WhileOp::getCanonicalizationPatterns(patterns, patterns.getContext());
 }

mlir/test/lib/Dialect/SCF/TestUpliftWhileToFor.cpp

@@ -34,7 +34,6 @@ struct TestSCFUpliftWhileToFor
     MLIRContext *ctx = op->getContext();
     RewritePatternSet patterns(ctx);
     scf::populateUpliftWhileToForPatterns(patterns);
-    scf::WhileOp::getCanonicalizationPatterns(patterns, ctx);
     if (failed(applyPatternsGreedily(op, std::move(patterns))))
       signalPassFailure();
   }