PR #20595: [nfc] Cleanup build files for simplifier transforms

Imported from GitHub PR #20595 This is part-2 of #18785. Motivation: Smaller build files, fewer merge conflicts, and convinience in development (more intuitive targets). For discussion surrounding this, check thread in #18785. Copybara import of the project: -- 8e54758 by Shraiysh Vaishay <svaishay@nvidia.com>: [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). Merging this change closes #20595 FUTURE_COPYBARA_INTEGRATE_REVIEW=#20595 from shraiysh:cleanup_simplifier_build_files 8e54758 PiperOrigin-RevId: 711650792
openxla · Jan 7, 2025 · a061947 · a061947
1 parent 422811e
commit a061947
Show file tree

Hide file tree

Showing 21 changed files with 2,115 additions and 2,082 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",