Skip to content

Commit

Permalink
PR #20595: [nfc] Cleanup build files for simplifier transforms
Browse files Browse the repository at this point in the history 
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
  • Loading branch information
@shraiysh @Google-ML-Automation
shraiysh authored and Google-ML-Automation committed Jan 7, 2025
1 parent e744bb4 commit 9d92fbb
Show file tree
Hide file tree
Showing 21 changed files with 2,122 additions and 2,084 deletions.
8 changes: 4 additions & 4 deletions xla/hlo/analysis/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -225,8 +225,8 @@ xla_cc_test(
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/testlib:hlo_hardware_independent_test_base",
"//xla/hlo/transforms:flatten_call_graph",
"//xla/hlo/transforms:hlo_dce",
"//xla/hlo/transforms/simplifiers:flatten_call_graph",
"//xla/hlo/transforms/simplifiers:hlo_dce",
"//xla/service:hlo_creation_utils",
"//xla/service:hlo_value",
"//xla/tsl/lib/core:status_test_util",
Expand Down Expand Up @@ -398,7 +398,7 @@ xla_cc_test(
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/testlib:hlo_hardware_independent_test_base",
"//xla/hlo/transforms:flatten_call_graph",
"//xla/hlo/transforms/simplifiers:flatten_call_graph",
"//xla/hlo/utils:hlo_matchers",
"//xla/service:hlo_buffer",
"//xla/service:hlo_value",
Expand Down Expand Up @@ -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",
Expand Down
2 changes: 1 addition & 1 deletion xla/hlo/evaluator/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
10 changes: 5 additions & 5 deletions xla/hlo/experimental/auto_sharding/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down
44 changes: 22 additions & 22 deletions xla/hlo/tools/hlo_opt/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
Loading

0 comments on commit 9d92fbb

Please sign in to comment.