[nfc] Cleanup build files for simplifier transforms

This is part-2 of #18785. Motivation: Smaller build files, fewer merge conflicts, and convinience in development (more intuitive targets).
openxla · Dec 16, 2024 · f9366d9 · f9366d9
1 parent a5b94d6
commit f9366d9
Show file tree

Hide file tree

Showing 20 changed files with 2,139 additions and 2,121 deletions.
diff --git a/xla/hlo/analysis/BUILD b/xla/hlo/analysis/BUILD
@@ -539,7 +539,7 @@ cc_library(
         "//xla:util",
         "//xla:xla_data_proto_cc",
         "//xla/hlo/ir:hlo",
-        "//xla/hlo/transforms:gather_simplifier",
+        "//xla/hlo/transforms/simplifiers:gather_simplifier",
         "//xla/hlo/utils:hlo_traversal",
         "//xla/service/gpu:matmul_indexing_utils",
         "@com_google_absl//absl/algorithm:container",

diff --git a/xla/hlo/evaluator/BUILD b/xla/hlo/evaluator/BUILD
@@ -139,7 +139,7 @@ xla_cc_test(
         "//xla/hlo/builder:xla_builder",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/testlib:hlo_hardware_independent_test_base",
-        "//xla/hlo/transforms:hlo_element_type_converter",
+        "//xla/hlo/transforms/simplifiers:hlo_element_type_converter",
         "//xla/service:call_graph",
         "//xla/service:dynamic_dimension_inference",
         "//xla/service:hlo_module_config",

diff --git a/xla/hlo/experimental/auto_sharding/BUILD b/xla/hlo/experimental/auto_sharding/BUILD
@@ -51,10 +51,10 @@ cc_library(
         "//xla/hlo/analysis:hlo_alias_analysis",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/pass:hlo_pass",
-        "//xla/hlo/transforms:hlo_constant_splitter",
-        "//xla/hlo/transforms:hlo_dce",
-        "//xla/hlo/transforms:hlo_memory_scheduler",
-        "//xla/hlo/transforms:optimize_input_output_buffer_alias",
+        "//xla/hlo/transforms/simplifiers:hlo_constant_splitter",
+        "//xla/hlo/transforms/simplifiers:hlo_dce",
+        "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler",
+        "//xla/hlo/transforms/simplifiers:optimize_input_output_buffer_alias",
         "//xla/hlo/utils:hlo_live_range",
         "//xla/hlo/utils:hlo_sharding_util",
         "//xla/service:buffer_value",
@@ -394,7 +394,7 @@ xla_cc_test(
         "//xla/hlo/parser:hlo_parser",
         "//xla/hlo/testlib:hlo_hardware_independent_test_base",
         "//xla/hlo/testlib:verified_hlo_module",
-        "//xla/hlo/transforms:hlo_memory_scheduler",
+        "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler",
         "//xla/hlo/utils:hlo_live_range",
         "//xla/hlo/utils:hlo_matchers",
         "//xla/service:buffer_value",