[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 · Jan 2, 2025 · 8e54758 · 8e54758
1 parent 2e8914d
commit 8e54758
Show file tree

Hide file tree

Showing 21 changed files with 2,095 additions and 2,077 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
@@ -140,7 +140,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",

diff --git a/xla/hlo/tools/hlo_opt/BUILD b/xla/hlo/tools/hlo_opt/BUILD
@@ -34,50 +34,50 @@ cc_library(
         "//xla/hlo/analysis:indexed_array_analysis",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/pass:hlo_pass_pipeline",
-        "//xla/hlo/transforms:all_reduce_folder",
-        "//xla/hlo/transforms:batch_dot_simplification",
-        "//xla/hlo/transforms:broadcast_canonicalizer",
         "//xla/hlo/transforms:cholesky_expander",
         "//xla/hlo/transforms:comparison_expander",
-        "//xla/hlo/transforms:conditional_canonicalizer",
         "//xla/hlo/transforms:convert_memory_placement_to_internal_annotations",
-        "//xla/hlo/transforms:convert_mover",
         "//xla/hlo/transforms:convolution_4d_expander",
-        "//xla/hlo/transforms:convolution_group_converter",
         "//xla/hlo/transforms:convolution_pred_expander",
         "//xla/hlo/transforms:dot_decomposer",
-        "//xla/hlo/transforms:dynamic_dimension_simplifier",
         "//xla/hlo/transforms:dynamic_index_splitter",
         "//xla/hlo/transforms:eigh_expander",
-        "//xla/hlo/transforms:flatten_call_graph",
-        "//xla/hlo/transforms:float_normalization",
-        "//xla/hlo/transforms:gather_simplifier",
-        "//xla/hlo/transforms:hlo_constant_folding",
-        "//xla/hlo/transforms:hlo_dce",
         "//xla/hlo/transforms:logistic_expander",
         "//xla/hlo/transforms:operand_upcaster",
         "//xla/hlo/transforms:optimization_barrier_expander",
-        "//xla/hlo/transforms:optimize_input_output_buffer_alias",
         "//xla/hlo/transforms:qr_expander",
         "//xla/hlo/transforms:real_imag_expander",
         "//xla/hlo/transforms:reduce_decomposer",
         "//xla/hlo/transforms:reshape_decomposer",
-        "//xla/hlo/transforms:reshape_mover",
-        "//xla/hlo/transforms:result_caster",
         "//xla/hlo/transforms:rng_expander",
-        "//xla/hlo/transforms:simplify_fp_conversions",
-        "//xla/hlo/transforms:slice_sinker",
-        "//xla/hlo/transforms:sort_simplifier",
         "//xla/hlo/transforms:stable_sort_expander",
         "//xla/hlo/transforms:stochastic_convert_decomposer",
-        "//xla/hlo/transforms:sub_byte_normalization",
-        "//xla/hlo/transforms:tree_reduction_rewriter",
-        "//xla/hlo/transforms:tuple_simplifier",
         "//xla/hlo/transforms:while_loop_trip_count_annotator",
-        "//xla/hlo/transforms:zero_sized_hlo_elimination",
         "//xla/hlo/transforms/collectives:all_gather_broadcast_reorder",
         "//xla/hlo/transforms/collectives:all_reduce_contiguous",
         "//xla/hlo/transforms/collectives:collective_quantizer",
+        "//xla/hlo/transforms/simplifiers:all_reduce_folder",
+        "//xla/hlo/transforms/simplifiers:batch_dot_simplification",
+        "//xla/hlo/transforms/simplifiers:broadcast_canonicalizer",
+        "//xla/hlo/transforms/simplifiers:conditional_canonicalizer",
+        "//xla/hlo/transforms/simplifiers:convert_mover",
+        "//xla/hlo/transforms/simplifiers:convolution_group_converter",
+        "//xla/hlo/transforms/simplifiers:dynamic_dimension_simplifier",
+        "//xla/hlo/transforms/simplifiers:flatten_call_graph",
+        "//xla/hlo/transforms/simplifiers:float_normalization",
+        "//xla/hlo/transforms/simplifiers:gather_simplifier",
+        "//xla/hlo/transforms/simplifiers:hlo_constant_folding",
+        "//xla/hlo/transforms/simplifiers:hlo_dce",
+        "//xla/hlo/transforms/simplifiers:optimize_input_output_buffer_alias",
+        "//xla/hlo/transforms/simplifiers:reshape_mover",
+        "//xla/hlo/transforms/simplifiers:result_caster",
+        "//xla/hlo/transforms/simplifiers:simplify_fp_conversions",
+        "//xla/hlo/transforms/simplifiers:slice_sinker",
+        "//xla/hlo/transforms/simplifiers:sort_simplifier",
+        "//xla/hlo/transforms/simplifiers:sub_byte_normalization",
+        "//xla/hlo/transforms/simplifiers:tree_reduction_rewriter",
+        "//xla/hlo/transforms/simplifiers:tuple_simplifier",
+        "//xla/hlo/transforms/simplifiers:zero_sized_hlo_elimination",
         "//xla/hlo/transforms/tests:dummy_passes",
         "//xla/service:float_support",
         "//xla/service:platform_util",