Update to match upstream API change (NFC).

This method was renamed but staging function kept, switch to renamed variant.

PiperOrigin-RevId: 713242393

Author: jpienaar

deallocation/transforms/buffer_reuse.cc

@@ -536,7 +536,7 @@ struct BufferReusePass : public impl::BufferReusePassBase<BufferReusePass> {
     eliminateCopies(block, /*root=*/block);
     do {
       // Eliminate dead code.
-      (void)applyPatternsAndFoldGreedily(getOperation(), {});
+      (void)applyPatternsGreedily(getOperation(), {});
       // Only coalesce dealloc/alloc pairs that are immediate neighbors, to
       // make sure we don't accidentally extend the live range of a buffer.
       result = reuseBuffers(block, BufferReuseMode::CONSERVATIVE);
@@ -547,7 +547,7 @@ struct BufferReusePass : public impl::BufferReusePassBase<BufferReusePass> {
     // Now we can also coalesce distant dealloc/alloc pairs.
     reuseBuffers(block, BufferReuseMode::AGGRESSIVE);
     promoteBuffers(block);
-    (void)applyPatternsAndFoldGreedily(getOperation(), {});
+    (void)applyPatternsGreedily(getOperation(), {});
   }
 };
 

mhlo/transforms/broadcast_propagation/broadcast_propagation.cc

@@ -439,8 +439,8 @@ struct BroadcastPropagationPass
     GreedyRewriteConfig config;
     config.useTopDownTraversal = false;
 
-    if (failed(applyPatternsAndFoldGreedily(getOperation(), std::move(patterns),
-                                            config))) {
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns),
+                                     config))) {
       return signalPassFailure();
     }
   }

mhlo/transforms/collapse_elementwise_map/collapse_elementwise_map.cc

@@ -92,8 +92,7 @@ struct CollapseElementwiseMapPass
     MLIRContext *ctx = &getContext();
     RewritePatternSet patterns(ctx);
     patterns.add<ConvertMapOfElementwiseOps>(ctx);
-    if (failed(
-            applyPatternsAndFoldGreedily(getOperation(), std::move(patterns))))
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns))))
       return signalPassFailure();
   }
 };

mhlo/transforms/legalize_dot_to_dot_general/legalize_dot_to_dot_general.cc

@@ -68,8 +68,7 @@ struct LegalizeDotToDotGeneralPass
   void runOnOperation() override {
     RewritePatternSet patterns(&getContext());
     populateDotToDotGeneralPatterns(&getContext(), &patterns);
-    if (failed(applyPatternsAndFoldGreedily(getOperation(),
-                                            std::move(patterns)))) {
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns)))) {
       return signalPassFailure();
     }
   }

mhlo/transforms/legalize_einsum_to_dot_general/legalize_einsum_to_dot_general.cc

@@ -179,8 +179,7 @@ struct LegalizeEinsumToDotGeneralPass
   void runOnOperation() override {
     RewritePatternSet patterns(&getContext());
     populateEinsumToDotGeneralPatterns(&getContext(), &patterns);
-    if (failed(applyPatternsAndFoldGreedily(getOperation(),
-                                            std::move(patterns)))) {
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns)))) {
       return signalPassFailure();
     }
   }

mhlo/transforms/legalize_torch_index_select_to_gather/legalize_torch_index_select_to_gather.cc

@@ -139,8 +139,7 @@ struct LegalizeTorchIndexSelectToGatherPass
   void runOnOperation() override {
     RewritePatternSet patterns(&getContext());
     populateTorchIndexSelectToGatherPatterns(&getContext(), &patterns);
-    if (failed(
-            applyPatternsAndFoldGreedily(getOperation(), std::move(patterns))))
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns))))
       return signalPassFailure();
   }
 };

mhlo/transforms/legalize_trigonometric_to_approximation/legalize_trigonometric_to_approximation.cc

@@ -172,8 +172,7 @@ struct LegalizeTrigonometricToApproximationPass
   void runOnOperation() override {
     RewritePatternSet patterns(&getContext());
     populateTrigonometricToApproximationPatterns(&getContext(), &patterns);
-    if (failed(applyPatternsAndFoldGreedily(getOperation(),
-                                            std::move(patterns)))) {
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns)))) {
       return signalPassFailure();
     }
   }

mhlo/transforms/merge_assuming_ops/merge_assuming_ops.cc

@@ -434,8 +434,8 @@ struct MergeAssumingOpsPass
     mhlo::populateMergeAssumingOpsPatterns(ctx, &patterns);
     GreedyRewriteConfig config;
     config.maxIterations = GreedyRewriteConfig::kNoLimit;
-    if (failed(applyPatternsAndFoldGreedily(getOperation(), std::move(patterns),
-                                            config))) {
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns),
+                                     config))) {
       return signalPassFailure();
     }
   }

mhlo/transforms/mhlo_flatten_tuple/mhlo_flatten_tuple.cc

@@ -132,8 +132,7 @@ class FlattenTuplePass : public impl::FlattenTuplePassBase<FlattenTuplePass> {
     MLIRContext *context = &getContext();
     RewritePatternSet patterns(context);
     patterns.add<FlattenCustomCallOp>(context);
-    if (failed(applyPatternsAndFoldGreedily(getOperation(),
-                                            std::move(patterns)))) {
+    if (failed(applyPatternsGreedily(getOperation(), std::move(patterns)))) {
       signalPassFailure();
     }
   }

mhlo/transforms/shape_simplification/shape_simplification.cc

@@ -242,7 +242,7 @@ struct ShapeSimplification
                  ExtractFromBroadcastedTensorCanonicalizationPattern>(context);
 
     auto func = getOperation();
-    if (failed(applyPatternsAndFoldGreedily(func, std::move(patterns))))
+    if (failed(applyPatternsGreedily(func, std::move(patterns))))
       return signalPassFailure();
   }
 };