Put thlo & gml_st on ice.

We can revert this when/if we need this. PiperOrigin-RevId: 585124862
ROCm · Nov 24, 2023 · 8744c9a · 8744c9a
1 parent acd327d
commit 8744c9a
Show file tree

Hide file tree

Showing 126 changed files with 35 additions and 19,822 deletions.
diff --git a/xla/mlir/backends/cpu/BUILD b/xla/mlir/backends/cpu/BUILD
@@ -1,5 +1,5 @@
-load("//xla:xla.bzl", "xla_cc_binary")
 load("@bazel_skylib//rules:build_test.bzl", "build_test")
+load("//xla:xla.bzl", "xla_cc_binary")
 
 package(
     # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@@ -21,12 +21,8 @@ xla_cc_binary(
         "//xla/mlir/backends/cpu/transforms:passes",
         "//xla/mlir/xla_cpu/ir:xla_cpu",
         "//xla/mlir_hlo:all_passes",
-        "//xla/mlir_hlo:gml_st",
-        "//xla/mlir_hlo:gml_st_passes",
-        "//xla/mlir_hlo:gml_st_test_passes",
         "//xla/mlir_hlo:hlo_dialect_registration",
         "//xla/mlir_hlo:lhlo",
-        "//xla/mlir_hlo:thlo",
         "//xla/service/cpu:cpu_compiler",
         "@llvm-project//mlir:BufferizationTransforms",
         "@llvm-project//mlir:FuncExtensions",

diff --git a/xla/mlir/backends/cpu/xla-cpu-opt.cc b/xla/mlir/backends/cpu/xla-cpu-opt.cc
@@ -24,28 +24,21 @@ limitations under the License.
 #include "stablehlo/dialect/Register.h"  // from @stablehlo
 #include "xla/mlir/backends/cpu/transforms/passes.h"
 #include "xla/mlir/xla_cpu/ir/xla_cpu.h"
-#include "xla/mlir_hlo/gml_st/IR/gml_st_ops.h"
-#include "xla/mlir_hlo/gml_st/transforms/passes.h"
-#include "xla/mlir_hlo/gml_st/transforms/test_passes.h"
 #include "xla/mlir_hlo/lhlo/IR/lhlo_ops.h"
 #include "xla/mlir_hlo/lhlo/transforms/passes.h"
 #include "xla/mlir_hlo/mhlo/IR/register.h"
 #include "xla/mlir_hlo/mhlo/transforms/passes.h"
-#include "xla/mlir_hlo/thlo/IR/thlo_ops.h"
 
 int main(int argc, char **argv) {
   mlir::mhlo::registerAllMhloPasses();
   mlir::lmhlo::registerAllLmhloPasses();
-  mlir::gml_st::registerGmlStPasses();
-  mlir::gml_st::registerGmlStTestPasses();
   mlir::bufferization::registerBufferizationPasses();
 
   mlir::DialectRegistry registry;
   mlir::mhlo::registerAllMhloDialects(registry);
   mlir::stablehlo::registerAllDialects(registry);
   registry.insert<mlir::func::FuncDialect, mlir::lmhlo::LmhloDialect,
                   mlir::memref::MemRefDialect, mlir::scf::SCFDialect,
-                  mlir::gml_st::GmlStDialect, mlir::thlo::THLODialect,
                   mlir::linalg::LinalgDialect, mlir::tensor::TensorDialect,
                   mlir::vector::VectorDialect, mlir::xla_cpu::XlaCpuDialect>();
   mlir::func::registerAllExtensions(registry);