rename pass.

openxla · Feb 6, 2024 · d1b68d7 · d1b68d7
1 parent fdc919a
commit d1b68d7
Show file tree

Hide file tree

Showing 7 changed files with 16 additions and 14 deletions.
diff --git a/stablehlo/tests/chlo/legalize_chlo_decomposition.mlir b/stablehlo/tests/chlo/legalize_chlo_decomposition.mlir
@@ -1,4 +1,4 @@
-// RUN: stablehlo-opt --pass-pipeline="builtin.module(func.func(stablehlo-legalize-chlo))" \
+// RUN: stablehlo-opt --pass-pipeline="builtin.module(func.func(chlo-legalize-to-stablehlo))" \
 // RUN:   --split-input-file --verify-diagnostics %s | FileCheck %s
 
 // CHECK-LABEL: func.func @asin_bf16(

diff --git a/stablehlo/tests/chlo/legalize_chlo_no_broadcast.mlir b/stablehlo/tests/chlo/legalize_chlo_no_broadcast.mlir
@@ -1,4 +1,4 @@
-// RUN: stablehlo-opt --pass-pipeline="builtin.module(func.func(stablehlo-legalize-chlo),cse)" \
+// RUN: stablehlo-opt --pass-pipeline="builtin.module(func.func(chlo-legalize-to-stablehlo),cse)" \
 // RUN:   --split-input-file --verify-diagnostics %s | FileCheck %s
 
 // Check the non-broadcast case for each registered op, then just check a

diff --git a/stablehlo/tests/chlo/legalize_chlo_with_broadcast.mlir b/stablehlo/tests/chlo/legalize_chlo_with_broadcast.mlir
@@ -1,4 +1,4 @@
-// RUN: stablehlo-opt --pass-pipeline="builtin.module(func.func(stablehlo-legalize-chlo),cse)" \
+// RUN: stablehlo-opt --pass-pipeline="builtin.module(func.func(chlo-legalize-to-stablehlo),cse)" \
 // RUN:   --split-input-file --verify-diagnostics %s | FileCheck %s
 
 // Check the non-broadcast case for each registered op, then just check a

diff --git a/stablehlo/transforms/CMakeLists.txt b/stablehlo/transforms/CMakeLists.txt
@@ -30,8 +30,8 @@ add_mlir_dialect_library(StablehloPasses
   VhloLegalizeToStablehlo.cpp
   VhloToVersion.cpp
   ChloLegalizeToStablehlo.cpp
-  StablehloCanonicalize.cpp
-
+  StablehloAggressiveSimplification.cpp
+   
   DEPENDS
   PassesIncGen
   ChloDecompositionPatterns

diff --git a/stablehlo/transforms/ChloLegalizeToStablehlo.cpp b/stablehlo/transforms/ChloLegalizeToStablehlo.cpp
@@ -26,7 +26,7 @@
 namespace mlir {
 namespace stablehlo {
 
-#define GEN_PASS_DEF_LEGALIZECHLO
+#define GEN_PASS_DEF_CHLOLEGALIZETOSTABLEHLOPASS
 #include "stablehlo/transforms/Passes.h.inc"
 
 namespace {
@@ -2161,7 +2161,8 @@ struct ConvertZetaOp final : OpConversionPattern<mlir::chlo::ZetaOp> {
 // Pass Definition.
 //===----------------------------------------------------------------------===//
 
-struct LegalizeChlo final : impl::LegalizeChloBase<LegalizeChlo> {
+struct ChloLegalizeToStablehloPass final
+    : impl::ChloLegalizeToStablehloPassBase<ChloLegalizeToStablehloPass> {
   void getDependentDialects(DialectRegistry &registry) const override {
     registry.insert<mlir::scf::SCFDialect, mlir::shape::ShapeDialect,
                     mlir::stablehlo::StablehloDialect,

diff --git a/stablehlo/transforms/Passes.td b/stablehlo/transforms/Passes.td
@@ -97,12 +97,12 @@ def VhloToVersionPass : Pass<"vhlo-to-version"> {
   ];
 }
 
-def LegalizeChlo :
-    InterfacePass<"stablehlo-legalize-chlo", "mlir::FunctionOpInterface"> {
+def ChloLegalizeToStablehloPass :
+    InterfacePass<"chlo-legalize-to-stablehlo", "mlir::FunctionOpInterface"> {
   let summary = "Legalizes from CHLO ops flow to StableHLO and Shape ops";
 }
 
-def StableHLOCanonicalize :
-    InterfacePass<"stablehlo-canonicalize", "mlir::FunctionOpInterface"> {
+def StablehloAggressiveSimplificationPass :
+    InterfacePass<"stablehlo-aggressive-simplification", "mlir::FunctionOpInterface"> {
   let summary = "Canonicalizes StableHLO operations";
 }
diff --git a/...ehlo/transforms/StablehloCanonicalize.cpp → ...rms/StablehloAggressiveSimplification.cpp b/...ehlo/transforms/StablehloCanonicalize.cpp → ...rms/StablehloAggressiveSimplification.cpp
@@ -32,7 +32,7 @@
 namespace mlir {
 namespace stablehlo {
 
-#define GEN_PASS_DEF_STABLEHLOCANONICALIZE
+#define GEN_PASS_DEF_STABLEHLOAGGRESSIVESIMPLIFICATIONPASS
 #include "stablehlo/transforms/Passes.h.inc"
 
 namespace {
@@ -1129,8 +1129,9 @@ struct ReorderElementwiseAndShapeOp final
   }
 };
 
-struct StableHLOCanonicalize final
-    : impl::StableHLOCanonicalizeBase<StableHLOCanonicalize> {
+struct StablehloAggressiveSimplificationPass final
+    : impl::StablehloAggressiveSimplificationPassBase<
+          StablehloAggressiveSimplificationPass> {
   void runOnOperation() override {
     MLIRContext *ctx = &getContext();
     RewritePatternSet patterns(ctx);