mhlo removed 3 passes:

- mhlo-legalize-control-flow
- mhlo-legalize-to-std
- hlo-legalize-sort

Do they have/need replacements?...

Author: paul0403

frontend/catalyst/pipelines.py

@@ -184,10 +184,10 @@ def get_hlo_lowering_stage(_options: CompileOptions) -> List[str]:
         "canonicalize",
         "func.func(chlo-legalize-to-hlo)",
         "stablehlo-legalize-to-hlo",
-        "func.func(mhlo-legalize-control-flow)",
+        # "func.func(mhlo-legalize-control-flow)",
         "func.func(hlo-legalize-to-linalg)",
-        "func.func(mhlo-legalize-to-std)",
-        "func.func(hlo-legalize-sort)",
+        # "func.func(mhlo-legalize-to-std)",
+        # "func.func(hlo-legalize-sort)",
         "convert-to-signless",
         "canonicalize",
         "scatter-lowering",

mlir/lib/Driver/Pipelines.cpp

@@ -42,10 +42,10 @@ void createHloLoweringPipeline(OpPassManager &pm)
     pm.addPass(mlir::createCanonicalizerPass());
     pm.addNestedPass<mlir::func::FuncOp>(mhlo::createChloLegalizeToHloPass());
     pm.addPass(mlir::mhlo::createStablehloLegalizeToHloPass());
-    pm.addNestedPass<mlir::func::FuncOp>(mhlo::createLegalizeControlFlowPass());
+    // pm.addNestedPass<mlir::func::FuncOp>(mhlo::createLegalizeControlFlowPass());
     pm.addNestedPass<mlir::func::FuncOp>(mhlo::createLegalizeHloToLinalgPass());
-    pm.addNestedPass<mlir::func::FuncOp>(mhlo::createLegalizeToStdPass());
-    pm.addNestedPass<mlir::func::FuncOp>(mhlo::createLegalizeSortPass());
+    // pm.addNestedPass<mlir::func::FuncOp>(mhlo::createLegalizeToStdPass());
+    // pm.addNestedPass<mlir::func::FuncOp>(mhlo::createLegalizeSortPass());
     pm.addPass(mlir::mhlo::createConvertToSignlessPass());
     pm.addPass(mlir::createCanonicalizerPass());
     pm.addPass(catalyst::createScatterLoweringPass());