From a061947bac15a8a82dd517e20a475214d7bf5c4c Mon Sep 17 00:00:00 2001 From: Shraiysh Date: Fri, 3 Jan 2025 00:02:09 -0800 Subject: [PATCH] PR #20595: [nfc] Cleanup build files for simplifier transforms Imported from GitHub PR https://github.com/openxla/xla/pull/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: -- 8e547580014fc3686e4bbaeadd4561de36f426df by Shraiysh Vaishay : [nfc] Cleanup build files for simplifier transforms This is part-2 of https://github.com/openxla/xla/pull/18785. Motivation: Smaller build files, fewer merge conflicts, and convinience in development (more intuitive targets). Merging this change closes #20595 FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/20595 from shraiysh:cleanup_simplifier_build_files 8e547580014fc3686e4bbaeadd4561de36f426df PiperOrigin-RevId: 711650792 --- xla/hlo/analysis/BUILD | 2 +- xla/hlo/evaluator/BUILD | 2 +- xla/hlo/experimental/auto_sharding/BUILD | 10 +- xla/hlo/tools/hlo_opt/BUILD | 44 +- xla/hlo/transforms/BUILD | 2124 +++------------------ xla/hlo/transforms/collectives/BUILD | 8 +- xla/hlo/transforms/simplifiers/BUILD | 1535 +++++++++++++++ xla/python/BUILD | 6 +- xla/service/BUILD | 226 +-- xla/service/cpu/BUILD | 46 +- xla/service/gpu/BUILD | 112 +- xla/service/gpu/autotuning/BUILD | 4 +- xla/service/gpu/fusions/triton/BUILD | 2 +- xla/service/gpu/transforms/BUILD | 38 +- xla/service/memory_space_assignment/BUILD | 2 +- xla/service/spmd/BUILD | 6 +- xla/service/spmd/shardy/BUILD | 4 +- xla/tests/BUILD | 10 +- xla/tools/BUILD | 6 +- xla/tools/hlo_bisect/BUILD | 2 +- xla/tools/hlo_opt/BUILD | 8 +- 21 files changed, 2115 insertions(+), 2082 deletions(-) create mode 100644 xla/hlo/transforms/simplifiers/BUILD diff --git a/xla/hlo/analysis/BUILD b/xla/hlo/analysis/BUILD index 6971952650f9d4..8bdc576b505905 100644 --- 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 index ef2a6b7fb7603c..b9a3c348ef30be 100644 --- 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 index df8aaf63583c6d..8e9d0cf51234ee 100644 --- 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 index 9af5f2251d5d64..490019fa0e96a4 100644 --- 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", diff --git a/xla/hlo/transforms/BUILD b/xla/hlo/transforms/BUILD index e4b9b8d85d9de3..4450f63f57fcc2 100644 --- a/xla/hlo/transforms/BUILD +++ b/xla/hlo/transforms/BUILD @@ -2,12 +2,7 @@ # Implementation of XLA’s HLO transformations. load("//xla:xla.bzl", "xla_cc_test") -load("//xla/tsl:tsl.bzl", "tsl_copts") load("//xla/tsl/platform:rules_cc.bzl", "cc_library") -load( - "//xla/tsl/platform/default:cuda_build_defs.bzl", - "if_cuda_is_configured", -) package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], @@ -23,1716 +18,600 @@ package_group( ) cc_library( - name = "hlo_constant_splitter", - srcs = ["simplifiers/hlo_constant_splitter.cc"], - hdrs = ["simplifiers/hlo_constant_splitter.h"], + name = "bfloat16_propagation", + srcs = ["bfloat16_propagation.cc"], + hdrs = ["bfloat16_propagation.h"], deps = [ + "//xla:literal", + "//xla:shape_tree", + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/analysis:hlo_dataflow_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", + "//xla/service:float_support", + "//xla/service:hlo_value", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/cleanup", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/functional:function_ref", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", "@tsl//tsl/platform:statusor", ], ) xla_cc_test( - name = "hlo_constant_splitter_test", - srcs = ["simplifiers/hlo_constant_splitter_test.cc"], + name = "bfloat16_propagation_test", + srcs = ["bfloat16_propagation_test.cc"], deps = [ - ":hlo_constant_splitter", - ":hlo_dce", + ":bfloat16_propagation", + "//xla:comparison_util", + "//xla:literal_util", + "//xla:shape_util", "//xla:test", - "//xla:util", + "//xla:test_helpers", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/tsl/lib/core:status_test_util", + "//xla/service:float_support", + "//xla/service:hlo_verifier", + "//xla/tests:literal_test_util", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status:statusor", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", + "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) cc_library( - name = "all_reduce_folder", - srcs = ["simplifiers/all_reduce_folder.cc"], - hdrs = ["simplifiers/all_reduce_folder.h"], + name = "op_expander_pass", + srcs = ["expanders/op_expander_pass.cc"], + hdrs = ["expanders/op_expander_pass.h"], deps = [ - "//xla:xla_data_proto_cc", + "//xla:util", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/utils:hlo_query", - "//xla/service:all_reduce_key", "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - ], -) - -xla_cc_test( - name = "all_reduce_folder_test", - srcs = ["simplifiers/all_reduce_folder_test.cc"], - deps = [ - ":all_reduce_folder", - "//xla:test", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "broadcast_canonicalizer", - srcs = ["simplifiers/broadcast_canonicalizer.cc"], - hdrs = ["simplifiers/broadcast_canonicalizer.h"], + name = "optimization_barrier_expander", + srcs = ["expanders/optimization_barrier_expander.cc"], + hdrs = ["expanders/optimization_barrier_expander.h"], deps = [ - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:hlo_creation_utils", + ":op_expander_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "broadcast_canonicalizer_test", - srcs = ["simplifiers/broadcast_canonicalizer_test.cc"], - deps = [ - ":broadcast_canonicalizer", - "//xla:test", - "//xla:test_helpers", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) cc_library( - name = "bfloat16_conversion_folding", - srcs = ["simplifiers/bfloat16_conversion_folding.cc"], - hdrs = ["simplifiers/bfloat16_conversion_folding.h"], + name = "comparison_expander", + srcs = ["expanders/comparison_expander.cc"], + hdrs = ["expanders/comparison_expander.h"], deps = [ + ":op_expander_pass", + "//xla:comparison_util", + "//xla:literal_util", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", - "//xla/hlo/analysis:hlo_dataflow_analysis", "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:float_support", - "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:status", + "@com_google_absl//absl/types:span", ], ) -xla_cc_test( - name = "bfloat16_conversion_folding_test", - srcs = ["simplifiers/bfloat16_conversion_folding_test.cc"], +cc_library( + name = "cholesky_expander", + srcs = ["expanders/cholesky_expander.cc"], + hdrs = ["expanders/cholesky_expander.h"], deps = [ - ":bfloat16_conversion_folding", + ":op_expander_pass", + "//xla:literal", "//xla:shape_util", - "//xla:test", - "//xla:test_helpers", + "//xla:status_macros", + "//xla:util", "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:float_support", + "//xla/hlo/builder:xla_builder", + "//xla/hlo/builder/lib:arithmetic", + "//xla/hlo/builder/lib:constants", + "//xla/hlo/builder/lib:loops", + "//xla/hlo/builder/lib:math", + "//xla/hlo/builder/lib:matrix", + "//xla/hlo/builder/lib:slicing", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/status:statusor", - "@tsl//tsl/platform:test_main", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", ], ) cc_library( - name = "float_normalization", - srcs = ["simplifiers/float_normalization.cc"], - hdrs = ["simplifiers/float_normalization.h"], + name = "qr_expander", + srcs = ["expanders/qr_expander.cc"], + hdrs = ["expanders/qr_expander.h"], deps = [ - ":hlo_dce", - ":tuple_simplifier", + ":op_expander_pass", + "//xla:literal", "//xla:shape_util", + "//xla:status_macros", "//xla:util", "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:call_graph", - "//xla/service:float_support", + "//xla/hlo/builder:xla_builder", + "//xla/hlo/builder/lib:arithmetic", + "//xla/hlo/builder/lib:constants", + "//xla/hlo/builder/lib:loops", + "//xla/hlo/builder/lib:math", + "//xla/hlo/builder/lib:matrix", + "//xla/hlo/builder/lib:qr", + "//xla/hlo/builder/lib:slicing", + "//xla/service:hlo_creation_utils", "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:statusor", + ], +) + +cc_library( + name = "real_imag_expander", + srcs = ["expanders/real_imag_expander.cc"], + hdrs = ["expanders/real_imag_expander.h"], + deps = [ + ":op_expander_pass", + "//xla:literal_util", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", ], ) xla_cc_test( - name = "float_normalization_test", - srcs = ["simplifiers/float_normalization_test.cc"], + name = "real_imag_expander_test", + size = "small", + srcs = ["expanders/real_imag_expander_test.cc"], deps = [ - ":float_normalization", + ":real_imag_expander", + "//xla:literal", "//xla:shape_util", "//xla:test", - "//xla:test_helpers", - "//xla:xla_data_proto_cc", + "//xla:types", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:float_support", + "//xla/hlo/utils:hlo_matchers", "//xla/service:hlo_creation_utils", - "//xla/service:hlo_verifier", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", + "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "bfloat16_propagation", - srcs = ["bfloat16_propagation.cc"], - hdrs = ["bfloat16_propagation.h"], + name = "eigh_expander", + srcs = ["expanders/eigh_expander.cc"], + hdrs = ["expanders/eigh_expander.h"], deps = [ - ":hlo_dce", - ":tuple_simplifier", - "//xla:literal", - "//xla:shape_tree", + ":op_expander_pass", + "//xla:literal_util", "//xla:shape_util", + "//xla:status_macros", "//xla:util", "//xla:xla_data_proto_cc", - "//xla/hlo/analysis:hlo_dataflow_analysis", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:float_support", - "//xla/service:hlo_value", + "//xla/hlo/builder:xla_builder", + "//xla/hlo/builder/lib:arithmetic", + "//xla/hlo/builder/lib:comparators", + "//xla/hlo/builder/lib:constants", + "//xla/hlo/builder/lib:loops", + "//xla/hlo/builder/lib:math", + "//xla/hlo/builder/lib:matrix", + "//xla/hlo/builder/lib:slicing", + "//xla/service:hlo_creation_utils", "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/cleanup", "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:statusor", ], ) -xla_cc_test( - name = "bfloat16_propagation_test", - srcs = ["bfloat16_propagation_test.cc"], +cc_library( + name = "convolution_4d_expander", + srcs = ["expanders/convolution_4d_expander.cc"], + hdrs = ["expanders/convolution_4d_expander.h"], deps = [ - ":bfloat16_propagation", - "//xla:comparison_util", - "//xla:literal_util", + ":op_expander_pass", "//xla:shape_util", - "//xla:test", - "//xla:test_helpers", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:float_support", - "//xla/service:hlo_verifier", - "//xla/tests:literal_test_util", - "@com_google_absl//absl/log", + "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status:statusor", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep + "@com_google_absl//absl/strings", ], ) -cc_library( - name = "flatten_call_graph", - srcs = ["simplifiers/flatten_call_graph.cc"], - hdrs = ["simplifiers/flatten_call_graph.h"], +xla_cc_test( + name = "convolution_4d_expander_test", + srcs = ["expanders/convolution_4d_expander_test.cc"], deps = [ - "//xla:util", + "convolution_4d_expander", + "//xla:test", "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/hlo/utils:hlo_query", - "//xla/service:call_graph", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "hlo_computation_deduplicator", - srcs = ["simplifiers/hlo_computation_deduplicator.cc"], - hdrs = ["simplifiers/hlo_computation_deduplicator.h"], + name = "convolution_pred_expander", + srcs = ["expanders/convolution_pred_expander.cc"], + hdrs = ["expanders/convolution_pred_expander.h"], deps = [ + ":op_expander_pass", "//xla:shape_util", - "//xla:status_macros", - "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:logging", - ], -) - -xla_cc_test( - name = "hlo_computation_deduplicator_test", - size = "small", - srcs = ["simplifiers/hlo_computation_deduplicator_test.cc"], - deps = [ - ":hlo_computation_deduplicator", - "//xla:literal_util", - "//xla:shape_util", - "//xla:test", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", - ], -) - -xla_cc_test( - name = "flatten_call_graph_test", - srcs = ["simplifiers/flatten_call_graph_test.cc"], - deps = [ - ":flatten_call_graph", - "//xla:comparison_util", - "//xla:literal_util", - "//xla:shape_util", - "//xla:test", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:call_graph", - "@com_google_absl//absl/status:statusor", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "hlo_memory_scheduler", - srcs = ["simplifiers/hlo_memory_scheduler.cc"], - hdrs = ["simplifiers/hlo_memory_scheduler.h"], - local_defines = if_cuda_is_configured(["GOOGLE_CUDA=1"]), - deps = [ - "//xla:shape_util", - "//xla:status_macros", - "//xla:types", - "//xla:util", - "//xla/hlo/analysis:hlo_alias_analysis", - "//xla/hlo/analysis:tuple_points_to_analysis", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:buffer_value", - "//xla/service:logical_buffer", - "//xla/service/heap_simulator", - "//xla/tsl/lib/gtl:map_util", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:numbers", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/profiler/lib:scoped_annotation", - ], -) - -xla_cc_test( - name = "hlo_memory_scheduler_test", - srcs = ["simplifiers/hlo_memory_scheduler_test.cc"], - deps = [ - ":hlo_dce", - ":hlo_memory_scheduler", - "//xla:literal_util", - "//xla:shape_util", - "//xla:types", - "//xla:xla_data_proto_cc", - "//xla/hlo/analysis:hlo_alias_analysis", - "//xla/hlo/analysis:hlo_ordering", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:buffer_value", - "//xla/service:hlo_value", - "//xla/service:logical_buffer", - "//xla/service/heap_simulator", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/status", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "op_expander_pass", - srcs = ["expanders/op_expander_pass.cc"], - hdrs = ["expanders/op_expander_pass.h"], - deps = [ - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:statusor", - ], -) - -cc_library( - name = "optimization_barrier_expander", - srcs = ["expanders/optimization_barrier_expander.cc"], - hdrs = ["expanders/optimization_barrier_expander.h"], - deps = [ - ":op_expander_pass", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - ], -) - -cc_library( - name = "comparison_expander", - srcs = ["expanders/comparison_expander.cc"], - hdrs = ["expanders/comparison_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:comparison_util", - "//xla:literal_util", - "//xla:shape_util", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - ], -) - -cc_library( - name = "cholesky_expander", - srcs = ["expanders/cholesky_expander.cc"], - hdrs = ["expanders/cholesky_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:literal", - "//xla:shape_util", - "//xla:status_macros", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/builder:xla_builder", - "//xla/hlo/builder/lib:arithmetic", - "//xla/hlo/builder/lib:constants", - "//xla/hlo/builder/lib:loops", - "//xla/hlo/builder/lib:math", - "//xla/hlo/builder/lib:matrix", - "//xla/hlo/builder/lib:slicing", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - ], -) - -cc_library( - name = "qr_expander", - srcs = ["expanders/qr_expander.cc"], - hdrs = ["expanders/qr_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:literal", - "//xla:shape_util", - "//xla:status_macros", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/builder:xla_builder", - "//xla/hlo/builder/lib:arithmetic", - "//xla/hlo/builder/lib:constants", - "//xla/hlo/builder/lib:loops", - "//xla/hlo/builder/lib:math", - "//xla/hlo/builder/lib:matrix", - "//xla/hlo/builder/lib:qr", - "//xla/hlo/builder/lib:slicing", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - ], -) - -cc_library( - name = "real_imag_expander", - srcs = ["expanders/real_imag_expander.cc"], - hdrs = ["expanders/real_imag_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:literal_util", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - ], -) - -xla_cc_test( - name = "real_imag_expander_test", - size = "small", - srcs = ["expanders/real_imag_expander_test.cc"], - deps = [ - ":real_imag_expander", - "//xla:literal", - "//xla:shape_util", - "//xla:test", - "//xla:types", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "//xla/service:hlo_creation_utils", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "//xla/tsl/lib/core:status_test_util", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "eigh_expander", - srcs = ["expanders/eigh_expander.cc"], - hdrs = ["expanders/eigh_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:literal_util", - "//xla:shape_util", - "//xla:status_macros", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/builder:xla_builder", - "//xla/hlo/builder/lib:arithmetic", - "//xla/hlo/builder/lib:comparators", - "//xla/hlo/builder/lib:constants", - "//xla/hlo/builder/lib:loops", - "//xla/hlo/builder/lib:math", - "//xla/hlo/builder/lib:matrix", - "//xla/hlo/builder/lib:slicing", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - ], -) - -cc_library( - name = "convolution_4d_expander", - srcs = ["expanders/convolution_4d_expander.cc"], - hdrs = ["expanders/convolution_4d_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - ], -) - -xla_cc_test( - name = "convolution_4d_expander_test", - srcs = ["expanders/convolution_4d_expander_test.cc"], - deps = [ - "convolution_4d_expander", - "//xla:test", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "convolution_pred_expander", - srcs = ["expanders/convolution_pred_expander.cc"], - hdrs = ["expanders/convolution_pred_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/service:hlo_creation_utils", - "//xla/service:pattern_matcher", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - ], -) - -xla_cc_test( - name = "convolution_pred_expander_test", - srcs = ["expanders/convolution_pred_expander_test.cc"], - deps = [ - ":convolution_pred_expander", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "algebraic_simplifier", - srcs = ["simplifiers/algebraic_simplifier.cc"], - hdrs = ["simplifiers/algebraic_simplifier.h"], - copts = tsl_copts(), - deps = [ - "//xla:comparison_util", - "//xla:literal", - "//xla:literal_util", - "//xla:permutation_util", - "//xla:shape_util", - "//xla:status_macros", - "//xla:util", - "//xla:window_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/evaluator:hlo_evaluator", - "//xla/hlo/ir:hlo", - "//xla/hlo/ir:hlo_instruction_utils", - "//xla/hlo/pass:hlo_pass", - "//xla/hlo/utils:hlo_sharding_util", - "//xla/service:gather_scatter_utils", - "//xla/service:hlo_cost_analysis", - "//xla/service:hlo_creation_utils", - "//xla/service:hlo_module_config", - "//xla/service:host_memory_offload_annotations_hdr", - "//xla/service:host_offload_utils", - "//xla/service:pattern_matcher", - "//xla/service:shape_inference", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/numeric:bits", - "@com_google_absl//absl/status", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:status", - "@tsl//tsl/platform:statusor", - ], -) - -cc_library( - name = "tree_reduction_rewriter", - srcs = ["simplifiers/tree_reduction_rewriter.cc"], - hdrs = ["simplifiers/tree_reduction_rewriter.h"], - deps = [ - "//xla:shape_util", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/builder:padding", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:shape_inference", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:status", - ], -) - -xla_cc_test( - name = "algebraic_simplifier_test", - srcs = ["simplifiers/algebraic_simplifier_test.cc"], - deps = [ - ":algebraic_simplifier", - ":hlo_constant_folding", - "//xla:comparison_util", - "//xla:literal", - "//xla:literal_util", - "//xla:shape_util", - "//xla:test", - "//xla:util", - "//xla:window_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/pass:hlo_pass", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:hlo_creation_utils", - "//xla/service:host_memory_offload_annotations_hdr", - "//xla/service:layout_assignment", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "//xla/service:shape_inference", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/log", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/types:span", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "simplify_fp_conversions", - srcs = ["simplifiers/simplify_fp_conversions.cc"], - hdrs = ["simplifiers/simplify_fp_conversions.h"], - deps = [ - "//xla:comparison_util", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "simplify_fp_conversions_test", - srcs = ["simplifiers/simplify_fp_conversions_test.cc"], - deps = [ - ":simplify_fp_conversions", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_absl//absl/strings", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:status_matchers", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "logistic_expander", - srcs = ["expanders/logistic_expander.cc"], - hdrs = ["expanders/logistic_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:shape_util", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:logging", - ], -) - -xla_cc_test( - name = "logistic_expander_test", - srcs = ["expanders/logistic_expander_test.cc"], - deps = [ - ":logistic_expander", - "//xla:test", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:dynamic_padder", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "bitcast_dtypes_expander", - srcs = ["expanders/bitcast_dtypes_expander.cc"], - hdrs = ["expanders/bitcast_dtypes_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:literal_util", - "//xla:shape_util", - "//xla:status_macros", - "//xla:types", - "//xla:xla_data_proto_cc", - "//xla/hlo/builder:xla_builder", - "//xla/hlo/builder:xla_computation", - "//xla/hlo/builder/lib:arithmetic", - "//xla/hlo/builder/lib:broadcast", - "//xla/hlo/builder/lib:constants", - "//xla/hlo/ir:hlo", - "//xla/service:hlo_creation_utils", - "//xla/service:hlo_module_config", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "bitcast_dtypes_expander_test", - srcs = ["expanders/bitcast_dtypes_expander_test.cc"], - deps = [ - ":bitcast_dtypes_expander", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:filecheck", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "batch_dot_simplification", - srcs = ["simplifiers/batch_dot_simplification.cc"], - hdrs = ["simplifiers/batch_dot_simplification.h"], - deps = [ - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "batch_dot_simplification_test", - srcs = ["simplifiers/batch_dot_simplification_test.cc"], - deps = [ - ":batch_dot_simplification", - "//xla:test", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "convolution_group_converter", - srcs = ["simplifiers/convolution_group_converter.cc"], - hdrs = ["simplifiers/convolution_group_converter.h"], - deps = [ - "//xla:literal", - "//xla:literal_util", - "//xla:shape_util", - "//xla:status_macros", - "//xla:types", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:status", - ], -) - -xla_cc_test( - name = "convolution_group_converter_test", - size = "small", - srcs = ["simplifiers/convolution_group_converter_test.cc"], - deps = [ - ":convolution_group_converter", - "//xla:test", - "//xla:types", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "while_loop_trip_count_annotator", - srcs = ["while_loop_trip_count_annotator.cc"], - hdrs = ["while_loop_trip_count_annotator.h"], - deps = [ - "//xla:xla_data_proto_cc", - "//xla/hlo/analysis:while_loop_analysis", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - ], -) - -xla_cc_test( - name = "while_loop_trip_count_annotator_test", - srcs = ["while_loop_trip_count_annotator_test.cc"], - deps = [ - ":while_loop_trip_count_annotator", - "//xla:test", - "//xla:xla_data_proto_cc", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "defuser", - srcs = ["defuser.cc"], - hdrs = ["defuser.h"], - deps = [ - "//xla:status_macros", - "//xla:types", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:call_graph", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:status", - ], -) - -xla_cc_test( - name = "defuser_test", - srcs = ["defuser_test.cc"], - deps = [ - ":defuser", - "//xla:literal", - "//xla:literal_util", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", - ], -) - -xla_cc_test( - name = "despecializer_test", - srcs = ["despecializer_test.cc"], - deps = [ - ":despecializer", - "//xla:literal", - "//xla:shape_util", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_absl//absl/log", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "dot_decomposer", - srcs = ["expanders/dot_decomposer.cc"], - hdrs = ["expanders/dot_decomposer.h"], - deps = [ - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:shape_inference", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "dot_decomposer_test", - srcs = ["expanders/dot_decomposer_test.cc"], - deps = [ - ":dot_decomposer", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "dot_dimension_merger", - srcs = ["simplifiers/dot_dimension_merger.cc"], - hdrs = ["simplifiers/dot_dimension_merger.h"], - deps = [ - "//xla:shape_util", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "dot_dimension_merger_test", - srcs = ["simplifiers/dot_dimension_merger_test.cc"], - deps = [ - ":dot_dimension_merger", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "dot_merger", - srcs = ["simplifiers/dot_merger.cc"], - hdrs = ["simplifiers/dot_merger.h"], - deps = [ - "//xla:protobuf_util", - "//xla:shape_util", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:shape_inference", - "//xla/service/graphcycles", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "dot_merger_test", - srcs = ["simplifiers/dot_merger_test.cc"], - deps = [ - ":algebraic_simplifier", - ":dot_merger", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "convert_mover", - srcs = ["simplifiers/convert_mover.cc"], - hdrs = ["simplifiers/convert_mover.h"], - deps = [ - "//xla:literal", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", "//xla/service:hlo_creation_utils", + "//xla/service:pattern_matcher", "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", + "@com_google_absl//absl/strings", ], ) xla_cc_test( - name = "convert_mover_test", - srcs = ["simplifiers/convert_mover_test.cc"], + name = "convolution_pred_expander_test", + srcs = ["expanders/convolution_pred_expander_test.cc"], deps = [ - ":convert_mover", + ":convolution_pred_expander", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", - "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep - ], -) - -cc_library( - name = "tuple_simplifier", - srcs = ["simplifiers/tuple_simplifier.cc"], - hdrs = ["simplifiers/tuple_simplifier.h"], - deps = [ - "//xla:shape_util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "tuple_simplifier_test", - srcs = ["simplifiers/tuple_simplifier_test.cc"], - deps = [ - ":tuple_simplifier", - "//xla:shape_util", - "//xla:test", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "reshape_mover", - srcs = ["simplifiers/reshape_mover.cc"], - hdrs = ["simplifiers/reshape_mover.h"], + name = "logistic_expander", + srcs = ["expanders/logistic_expander.cc"], + hdrs = ["expanders/logistic_expander.h"], deps = [ - "//xla:permutation_util", + ":op_expander_pass", "//xla:shape_util", - "//xla:status_macros", "//xla:util", - "//xla/hlo/pass:hlo_pass", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - ], -) - -cc_library( - name = "reshape_decomposer", - srcs = ["expanders/reshape_decomposer.cc"], - hdrs = ["expanders/reshape_decomposer.h"], - deps = [ - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - ], -) - -cc_library( - name = "reduce_decomposer", - srcs = ["expanders/reduce_decomposer.cc"], - hdrs = ["expanders/reduce_decomposer.h"], - deps = [ + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:logging", ], ) xla_cc_test( - name = "reduce_decomposer_test", - srcs = ["expanders/reduce_decomposer_test.cc"], - deps = [ - ":reduce_decomposer", - "//xla:test", - "//xla:test_helpers", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", - ], -) - -xla_cc_test( - name = "reshape_decomposer_test", - srcs = ["expanders/reshape_decomposer_test.cc"], + name = "logistic_expander_test", + srcs = ["expanders/logistic_expander_test.cc"], deps = [ - ":reshape_decomposer", + ":logistic_expander", "//xla:test", - "//xla:test_helpers", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_absl//absl/algorithm:container", + "//xla/service:dynamic_padder", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) cc_library( - name = "dynamic_dimension_simplifier", - srcs = ["simplifiers/dynamic_dimension_simplifier.cc"], - hdrs = ["simplifiers/dynamic_dimension_simplifier.h"], - deps = [ - "//xla:status_macros", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - ], -) - -xla_cc_test( - name = "dynamic_dimension_simplifier_test", - srcs = ["simplifiers/dynamic_dimension_simplifier_test.cc"], + name = "bitcast_dtypes_expander", + srcs = ["expanders/bitcast_dtypes_expander.cc"], + hdrs = ["expanders/bitcast_dtypes_expander.h"], deps = [ - ":dynamic_dimension_simplifier", - "//xla:literal", + ":op_expander_pass", + "//xla:literal_util", "//xla:shape_util", - "//xla:test", + "//xla:status_macros", "//xla:types", - "//xla:window_util", "//xla:xla_data_proto_cc", + "//xla/hlo/builder:xla_builder", + "//xla/hlo/builder:xla_computation", + "//xla/hlo/builder/lib:arithmetic", + "//xla/hlo/builder/lib:broadcast", + "//xla/hlo/builder/lib:constants", "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:hlo_creation_utils", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "//xla/service:shape_inference", - "//xla/tsl/lib/core:status_test_util", + "//xla/service:hlo_module_config", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", - "@tsl//tsl/platform:test_main", # fixdeps: keep + "@com_google_absl//absl/strings:str_format", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:statusor", ], ) xla_cc_test( - name = "reshape_mover_test", - srcs = ["simplifiers/reshape_mover_test.cc"], + name = "bitcast_dtypes_expander_test", + srcs = ["expanders/bitcast_dtypes_expander_test.cc"], deps = [ - ":algebraic_simplifier", - ":reshape_mover", + ":bitcast_dtypes_expander", "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", + "//xla/hlo/testlib:filecheck", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:hlo_verifier", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", + "//xla/hlo/utils:hlo_matchers", "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/status", - "@com_google_googletest//:gtest", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "memory_space_propagation", - srcs = ["memory_space_propagation.cc"], - hdrs = ["memory_space_propagation.h"], + name = "while_loop_trip_count_annotator", + srcs = ["while_loop_trip_count_annotator.cc"], + hdrs = ["while_loop_trip_count_annotator.h"], deps = [ - "//xla:shape_util", - "//xla/hlo/analysis:hlo_dataflow_analysis", + "//xla:xla_data_proto_cc", + "//xla/hlo/analysis:while_loop_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", ], ) xla_cc_test( - name = "memory_space_propagation_test", - srcs = ["memory_space_propagation_test.cc"], + name = "while_loop_trip_count_annotator_test", + srcs = ["while_loop_trip_count_annotator_test.cc"], deps = [ - ":memory_space_propagation", - "//xla/hlo/parser:hlo_parser", + ":while_loop_trip_count_annotator", + "//xla:test", + "//xla:xla_data_proto_cc", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/hash", - "@com_google_absl//absl/status", - "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "hlo_dce", - srcs = ["simplifiers/hlo_dce.cc"], - hdrs = ["simplifiers/hlo_dce.h"], - deps = [ - "//xla:shape_util", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:statusor", - ], -) - -cc_library( - name = "hlo_rematerialization", - srcs = ["simplifiers/hlo_rematerialization.cc"], - hdrs = ["simplifiers/hlo_rematerialization.h"], - deps = [ - ":hlo_dce", - "//xla:shape_util", - "//xla:status_macros", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/analysis:hlo_dataflow_analysis", - "//xla/hlo/analysis:tuple_points_to_analysis", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/hlo/utils:hlo_query", - "//xla/service:call_graph", - "//xla/service:hlo_cost_analysis", - "//xla/service:logical_buffer", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/functional:function_ref", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:numbers", "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) cc_library( - name = "hlo_rematerialization_test_utils", - testonly = 1, - hdrs = ["simplifiers/hlo_rematerialization_test_utils.h"], - deps = [ - "//xla:literal_util", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@tsl//tsl/platform:test_main", - ], -) - -xla_cc_test( - name = "hlo_rematerialization_test_utils_test", - srcs = ["simplifiers/hlo_rematerialization_test_utils_test.cc"], + name = "defuser", + srcs = ["defuser.cc"], + hdrs = ["defuser.h"], deps = [ - ":hlo_rematerialization_test_utils", + "//xla:status_macros", + "//xla:types", + "//xla:util", "//xla/hlo/ir:hlo", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", + "//xla/hlo/pass:hlo_pass", + "//xla/service:call_graph", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:status", ], ) xla_cc_test( - name = "hlo_rematerialization_test", - srcs = ["simplifiers/hlo_rematerialization_test.cc"], + name = "defuser_test", + srcs = ["defuser_test.cc"], deps = [ - ":hlo_memory_scheduler", - ":hlo_rematerialization", - ":hlo_rematerialization_test_utils", + ":defuser", + "//xla:literal", + "//xla:literal_util", "//xla:shape_util", - "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", - "//xla/service:hlo_cost_analysis", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test_main", ], ) xla_cc_test( - name = "hlo_dce_test", - srcs = ["simplifiers/hlo_dce_test.cc"], + name = "despecializer_test", + srcs = ["despecializer_test.cc"], deps = [ - ":hlo_dce", - "//xla:literal_util", + ":despecializer", + "//xla:literal", "//xla:shape_util", - "//xla:types", - "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "//xla/tests:literal_test_util", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/types:span", + "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/log", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "hlo_constant_folding", - srcs = ["simplifiers/hlo_constant_folding.cc"], - hdrs = ["simplifiers/hlo_constant_folding.h"], + name = "dot_decomposer", + srcs = ["expanders/dot_decomposer.cc"], + hdrs = ["expanders/dot_decomposer.h"], deps = [ - "//xla:literal", "//xla:shape_util", - "//xla/hlo/evaluator:hlo_evaluator", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/service:slow_operation_alarm", + "//xla/service:shape_inference", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/time", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:statusor", ], ) xla_cc_test( - name = "hlo_constant_folding_test", - srcs = ["simplifiers/hlo_constant_folding_test.cc"], + name = "dot_decomposer_test", + srcs = ["expanders/dot_decomposer_test.cc"], deps = [ - ":hlo_constant_folding", - "//xla:literal", - "//xla:literal_util", - "//xla:permutation_util", - "//xla:shape_util", - "//xla:test", - "//xla:xla_data_proto_cc", + ":dot_decomposer", "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", + "@tsl//tsl/platform:test", + "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) -# copybara:uncomment_begin(google-only) -# xla_cc_test( -# name = "hlo_constant_folding_peak_heap_test", -# srcs = ["simplifiers/hlo_constant_folding_peak_heap_test.cc"], -# deps = [ -# ":hlo_constant_folding", -# "@com_google_googletest//:gtest", -# "@com_google_absl//absl/strings:str_format", -# "//xla:test", -# "//xla/hlo/testlib:hlo_hardware_independent_test_base", -# "@tsl//tsl/platform:statusor", -# "@tsl//tsl/platform:test_main", -# ], -# ) -# copybara:uncomment_end - cc_library( - name = "hlo_element_type_converter", - srcs = ["simplifiers/hlo_element_type_converter.cc"], - hdrs = ["simplifiers/hlo_element_type_converter.h"], + name = "reshape_decomposer", + srcs = ["expanders/reshape_decomposer.cc"], + hdrs = ["expanders/reshape_decomposer.h"], deps = [ - "//xla:literal", - "//xla:shape_util", - "//xla:types", - "//xla:xla_data_proto_cc", - "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/utils:hlo_query", + "//xla/service:hlo_creation_utils", "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - ], -) - -xla_cc_test( - name = "hlo_element_type_converter_test", - srcs = ["simplifiers/hlo_element_type_converter_test.cc"], - deps = [ - ":hlo_element_type_converter", - "//xla:xla_data_proto_cc", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "conditional_canonicalizer", - srcs = ["simplifiers/conditional_canonicalizer.cc"], - hdrs = ["simplifiers/conditional_canonicalizer.h"], + name = "reduce_decomposer", + srcs = ["expanders/reduce_decomposer.cc"], + hdrs = ["expanders/reduce_decomposer.h"], deps = [ - "//xla:status_macros", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", + "//xla/service:hlo_creation_utils", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", ], ) xla_cc_test( - name = "conditional_canonicalizer_test", - srcs = ["simplifiers/conditional_canonicalizer_test.cc"], + name = "reduce_decomposer_test", + srcs = ["expanders/reduce_decomposer_test.cc"], deps = [ - ":conditional_canonicalizer", - "//xla:shape_util", - "//xla:types", - "//xla:util", - "//xla/hlo/ir:hlo", + ":reduce_decomposer", + "//xla:test", + "//xla:test_helpers", "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "//xla/tests:literal_test_util", - "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", "@tsl//tsl/platform:test_main", ], ) -cc_library( - name = "zero_sized_hlo_elimination", - srcs = ["simplifiers/zero_sized_hlo_elimination.cc"], - hdrs = ["simplifiers/zero_sized_hlo_elimination.h"], - deps = [ - "//xla:literal", - "//xla:shape_util", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - ], -) - xla_cc_test( - name = "zero_sized_hlo_elimination_test", - srcs = ["simplifiers/zero_sized_hlo_elimination_test.cc"], + name = "reshape_decomposer_test", + srcs = ["expanders/reshape_decomposer_test.cc"], deps = [ - ":zero_sized_hlo_elimination", - "//xla:literal_util", - "//xla:shape_util", + ":reshape_decomposer", "//xla:test", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", + "//xla:test_helpers", + "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@tsl//tsl/platform:logging", - "@tsl//tsl/platform:statusor", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:test_main", ], ) cc_library( - name = "sort_simplifier", - srcs = ["simplifiers/sort_simplifier.cc"], - hdrs = ["simplifiers/sort_simplifier.h"], + name = "memory_space_propagation", + srcs = ["memory_space_propagation.cc"], + hdrs = ["memory_space_propagation.h"], deps = [ "//xla:shape_util", - "//xla:util", + "//xla/hlo/analysis:hlo_dataflow_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", ], ) xla_cc_test( - name = "sort_simplifier_test", - srcs = ["simplifiers/sort_simplifier_test.cc"], + name = "memory_space_propagation_test", + srcs = ["memory_space_propagation_test.cc"], deps = [ - ":sort_simplifier", - "//xla:test", + ":memory_space_propagation", "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", "//xla/tsl/lib/core:status_test_util", - "@tsl//tsl/platform:statusor", + "@com_google_absl//absl/hash", + "@com_google_absl//absl/status", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:test_main", ], ) @@ -1758,11 +637,11 @@ xla_cc_test( name = "stable_sort_expander_test", srcs = ["expanders/stable_sort_expander_test.cc"], deps = [ - ":algebraic_simplifier", ":stable_sort_expander", "//xla:test", "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/hlo/utils:hlo_matchers", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", @@ -1772,34 +651,6 @@ xla_cc_test( ], ) -cc_library( - name = "root_instruction_sinker", - srcs = ["simplifiers/root_instruction_sinker.cc"], - hdrs = ["simplifiers/root_instruction_sinker.h"], - deps = [ - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:tuple_util", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - ], -) - -xla_cc_test( - name = "root_instruction_sinker_test", - srcs = ["simplifiers/root_instruction_sinker_test.cc"], - deps = [ - ":root_instruction_sinker", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", - ], -) - cc_library( name = "convert_memory_placement_to_internal_annotations", srcs = ["convert_memory_placement_to_internal_annotations.cc"], @@ -1840,47 +691,6 @@ xla_cc_test( ], ) -cc_library( - name = "host_memory_transfer_asyncifier", - srcs = ["simplifiers/host_memory_transfer_asyncifier.cc"], - hdrs = ["simplifiers/host_memory_transfer_asyncifier.h"], - deps = [ - "//xla:shape_util", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "host_memory_transfer_asyncifier_test", - srcs = ["simplifiers/host_memory_transfer_asyncifier_test.cc"], - deps = [ - ":host_memory_transfer_asyncifier", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - cc_library( name = "host_offload_legalize", srcs = ["host_offload_legalize.cc"], @@ -2023,48 +833,14 @@ xla_cc_test( name = "host_offloading_prepare_test", srcs = ["host_offloading_prepare_test.cc"], deps = [ - ":host_offloading_prepare", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:host_memory_offload_annotations_hdr", - "//xla/tsl/lib/core:status_test_util", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "fusion_constant_sinking", - srcs = ["simplifiers/fusion_constant_sinking.cc"], - hdrs = ["simplifiers/fusion_constant_sinking.h"], - deps = [ - ":hlo_dce", - "//xla:shape_util", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "fusion_constant_sinking_test", - srcs = ["simplifiers/fusion_constant_sinking_test.cc"], - deps = [ - ":fusion_constant_sinking", - "//xla:test", + ":host_offloading_prepare", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", + "//xla/service:host_memory_offload_annotations_hdr", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test_main", ], @@ -2076,13 +852,13 @@ cc_library( hdrs = ["despecializer.h"], deps = [ ":defuser", - ":float_normalization", - ":hlo_memory_scheduler", - ":sub_byte_normalization", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", + "//xla/hlo/transforms/simplifiers:float_normalization", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:sub_byte_normalization", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -2127,73 +903,6 @@ xla_cc_test( ], ) -cc_library( - name = "optimize_input_output_buffer_alias", - srcs = ["simplifiers/optimize_input_output_buffer_alias.cc"], - hdrs = ["simplifiers/optimize_input_output_buffer_alias.h"], - deps = [ - "//xla:shape_util", - "//xla:status_macros", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - ], -) - -xla_cc_test( - name = "optimize_input_output_buffer_alias_test", - srcs = ["simplifiers/optimize_input_output_buffer_alias_test.cc"], - deps = [ - ":optimize_input_output_buffer_alias", - "//xla:shape_util", - "//xla:test", - "//xla:test_helpers", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_absl//absl/status", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "ar_crs_combiner", - srcs = ["simplifiers/ar_crs_combiner.cc"], - hdrs = ["simplifiers/ar_crs_combiner.h"], - deps = [ - "//xla:literal", - "//xla:literal_util", - "//xla:shape_util", - "//xla:status_macros", - "//xla:xla_data_proto_cc", - "//xla/hlo/analysis:hlo_replication_analysis", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/hlo/utils:hlo_query", - "//xla/service:call_graph", - "//xla/service:pattern_matcher", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:status", - "@tsl//tsl/platform:statusor", - ], -) - cc_library( name = "dynamic_index_splitter", srcs = ["expanders/dynamic_index_splitter.cc"], @@ -2227,65 +936,6 @@ xla_cc_test( ], ) -xla_cc_test( - name = "ar_crs_combiner_test", - srcs = ["simplifiers/ar_crs_combiner_test.cc"], - deps = [ - ":ar_crs_combiner", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/types:span", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "slice_sinker", - srcs = ["simplifiers/slice_sinker.cc"], - hdrs = ["simplifiers/slice_sinker.h"], - deps = [ - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - ], -) - -xla_cc_test( - name = "slice_sinker_test", - srcs = ["simplifiers/slice_sinker_test.cc"], - deps = [ - ":hlo_dce", - ":slice_sinker", - "//xla:literal_util", - "//xla:shape_util", - "//xla:types", - "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/service:pattern_matcher", - "//xla/service:pattern_matcher_gmock", - "//xla/tsl/lib/core:status_test_util", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - cc_library( name = "rng_expander", srcs = ["expanders/rng_expander.cc"], @@ -2365,142 +1015,6 @@ xla_cc_test( ], ) -cc_library( - name = "result_caster", - srcs = ["simplifiers/result_caster.cc"], - hdrs = ["simplifiers/result_caster.h"], - deps = [ - ":op_expander_pass", - "//xla:shape_util", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/service:shape_inference", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - ], -) - -xla_cc_test( - name = "result_caster_test", - srcs = ["simplifiers/result_caster_test.cc"], - deps = [ - ":result_caster", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_absl//absl/strings", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "convert_operand_folding", - srcs = ["simplifiers/convert_operand_folder.cc"], - hdrs = ["simplifiers/convert_operand_folder.h"], - deps = [ - ":op_expander_pass", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:errors", - ], -) - -xla_cc_test( - name = "convert_operand_folding_test", - srcs = ["simplifiers/convert_operand_folder_test.cc"], - deps = [ - ":convert_operand_folding", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_absl//absl/strings", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", - ], -) - -cc_library( - name = "instruction_hoister", - srcs = ["simplifiers/instruction_hoister.cc"], - hdrs = ["simplifiers/instruction_hoister.h"], - deps = [ - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:status", - ], -) - -cc_library( - name = "gather_simplifier", - srcs = ["simplifiers/gather_simplifier.cc"], - hdrs = ["simplifiers/gather_simplifier.h"], - deps = [ - ":op_expander_pass", - "//xla:literal_util", - "//xla:permutation_util", - "//xla:shape_util", - "//xla/hlo/ir:hlo", - "//xla/service:gather_scatter_utils", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:statusor", - ], -) - -cc_library( - name = "reduce_window_rewriter", - srcs = ["simplifiers/reduce_window_rewriter.cc"], - hdrs = ["simplifiers/reduce_window_rewriter.h"], - deps = [ - "//xla:shape_util", - "//xla:status_macros", - "//xla:util", - "//xla:window_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/log", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "reduce_window_rewriter_test", - srcs = ["simplifiers/reduce_window_rewriter_test.cc"], - deps = [ - ":reduce_window_rewriter", - "//xla:test", - "//xla:xla_data_proto_cc", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:test_main", - ], -) - cc_library( name = "stochastic_convert_decomposer", srcs = ["expanders/stochastic_convert_decomposer.cc"], @@ -2539,25 +1053,6 @@ xla_cc_test( ], ) -cc_library( - name = "sub_byte_normalization", - srcs = ["simplifiers/sub_byte_normalization.cc"], - hdrs = ["simplifiers/sub_byte_normalization.h"], - deps = [ - "//xla:shape_layout", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/status", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@tsl//tsl/platform:errors", - "@tsl//tsl/platform:status", - ], -) - cc_library( name = "sharding_format_picker", testonly = True, @@ -2575,18 +1070,6 @@ cc_library( ], ) -xla_cc_test( - name = "gather_simplifier_test", - srcs = ["simplifiers/gather_simplifier_test.cc"], - deps = [ - ":gather_simplifier", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", - ], -) - cc_library( name = "add_original_value", srcs = ["add_original_value.cc"], @@ -2624,3 +1107,18 @@ xla_cc_test( "@tsl//tsl/platform:test_main", ], ) + +alias( + name = "hlo_dce", + actual = "//xla/hlo/transforms/simplifiers:hlo_dce", +) + +alias( + name = "hlo_memory_scheduler", + actual = "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", +) + +alias( + name = "dynamic_dimension_simplifier", + actual = "//xla/hlo/transforms/simplifiers:dynamic_dimension_simplifier", +) diff --git a/xla/hlo/transforms/collectives/BUILD b/xla/hlo/transforms/collectives/BUILD index 809a163d27cf3b..b7f8aba244bdf0 100644 --- a/xla/hlo/transforms/collectives/BUILD +++ b/xla/hlo/transforms/collectives/BUILD @@ -40,7 +40,7 @@ xla_cc_test( ":all_gather_cse", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/hlo/utils:hlo_matchers", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings:string_view", @@ -364,7 +364,7 @@ cc_library( "//xla:shape_util", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log:check", @@ -404,8 +404,8 @@ cc_library( "//xla/hlo/analysis:hlo_ordering", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/service:call_graph", "//xla/service:tuple_util", "@com_google_absl//absl/container:flat_hash_set", diff --git a/xla/hlo/transforms/simplifiers/BUILD b/xla/hlo/transforms/simplifiers/BUILD new file mode 100644 index 00000000000000..e19f32f2cbbb7d --- /dev/null +++ b/xla/hlo/transforms/simplifiers/BUILD @@ -0,0 +1,1535 @@ +# Description: +# Implementation of XLA’s HLO simplifier transformations. + +load("//xla:xla.bzl", "xla_cc_test") +load("//xla/tsl:tsl.bzl", "tsl_copts") +load("//xla/tsl/platform:rules_cc.bzl", "cc_library") +load( + "//xla/tsl/platform/default:cuda_build_defs.bzl", + "if_cuda_is_configured", +) + +package( + # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], + default_visibility = [":friends"], + licenses = ["notice"], +) + +package_group( + name = "friends", + includes = [ + "//xla:friends", + ], +) + +cc_library( + name = "hlo_constant_splitter", + srcs = ["hlo_constant_splitter.cc"], + hdrs = ["hlo_constant_splitter.h"], + deps = [ + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/functional:function_ref", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "hlo_constant_splitter_test", + srcs = ["hlo_constant_splitter_test.cc"], + deps = [ + ":hlo_constant_splitter", + ":hlo_dce", + "//xla:test", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/tsl/lib/core:status_test_util", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "all_reduce_folder", + srcs = ["all_reduce_folder.cc"], + hdrs = ["all_reduce_folder.h"], + deps = [ + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/utils:hlo_query", + "//xla/service:all_reduce_key", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "all_reduce_folder_test", + srcs = ["all_reduce_folder_test.cc"], + deps = [ + ":all_reduce_folder", + "//xla:test", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "broadcast_canonicalizer", + srcs = ["broadcast_canonicalizer.cc"], + hdrs = ["broadcast_canonicalizer.h"], + deps = [ + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "broadcast_canonicalizer_test", + srcs = ["broadcast_canonicalizer_test.cc"], + deps = [ + ":broadcast_canonicalizer", + "//xla:test", + "//xla:test_helpers", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "bfloat16_conversion_folding", + srcs = ["bfloat16_conversion_folding.cc"], + hdrs = ["bfloat16_conversion_folding.h"], + deps = [ + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/analysis:hlo_dataflow_analysis", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:float_support", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:status", + ], +) + +xla_cc_test( + name = "bfloat16_conversion_folding_test", + srcs = ["bfloat16_conversion_folding_test.cc"], + deps = [ + ":bfloat16_conversion_folding", + "//xla:shape_util", + "//xla:test", + "//xla:test_helpers", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:float_support", + "@com_google_absl//absl/status:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "float_normalization", + srcs = ["float_normalization.cc"], + hdrs = ["float_normalization.h"], + deps = [ + ":hlo_dce", + ":tuple_simplifier", + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:call_graph", + "//xla/service:float_support", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "float_normalization_test", + srcs = ["float_normalization_test.cc"], + deps = [ + ":float_normalization", + "//xla:shape_util", + "//xla:test", + "//xla:test_helpers", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:float_support", + "//xla/service:hlo_creation_utils", + "//xla/service:hlo_verifier", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "hlo_computation_deduplicator", + srcs = ["hlo_computation_deduplicator.cc"], + hdrs = ["hlo_computation_deduplicator.h"], + deps = [ + "//xla:shape_util", + "//xla:status_macros", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:logging", + ], +) + +xla_cc_test( + name = "hlo_computation_deduplicator_test", + size = "small", + srcs = ["hlo_computation_deduplicator_test.cc"], + deps = [ + ":hlo_computation_deduplicator", + "//xla:literal_util", + "//xla:shape_util", + "//xla:test", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "flatten_call_graph", + srcs = ["flatten_call_graph.cc"], + hdrs = ["flatten_call_graph.h"], + deps = [ + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/utils:hlo_query", + "//xla/service:call_graph", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + ], +) + +xla_cc_test( + name = "flatten_call_graph_test", + srcs = ["flatten_call_graph_test.cc"], + deps = [ + ":flatten_call_graph", + "//xla:comparison_util", + "//xla:literal_util", + "//xla:shape_util", + "//xla:test", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:call_graph", + "@com_google_absl//absl/status:statusor", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "hlo_memory_scheduler", + srcs = ["hlo_memory_scheduler.cc"], + hdrs = ["hlo_memory_scheduler.h"], + local_defines = if_cuda_is_configured(["GOOGLE_CUDA=1"]), + deps = [ + "//xla:shape_util", + "//xla:status_macros", + "//xla:types", + "//xla:util", + "//xla/hlo/analysis:hlo_alias_analysis", + "//xla/hlo/analysis:tuple_points_to_analysis", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:buffer_value", + "//xla/service:logical_buffer", + "//xla/service/heap_simulator", + "//xla/tsl/lib/gtl:map_util", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:numbers", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/profiler/lib:scoped_annotation", + ], +) + +xla_cc_test( + name = "hlo_memory_scheduler_test", + srcs = ["hlo_memory_scheduler_test.cc"], + deps = [ + ":hlo_dce", + ":hlo_memory_scheduler", + "//xla:literal_util", + "//xla:shape_util", + "//xla:types", + "//xla:xla_data_proto_cc", + "//xla/hlo/analysis:hlo_alias_analysis", + "//xla/hlo/analysis:hlo_ordering", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:buffer_value", + "//xla/service:hlo_value", + "//xla/service:logical_buffer", + "//xla/service/heap_simulator", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/status", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "algebraic_simplifier", + srcs = ["algebraic_simplifier.cc"], + hdrs = ["algebraic_simplifier.h"], + copts = tsl_copts(), + deps = [ + "//xla:comparison_util", + "//xla:literal", + "//xla:literal_util", + "//xla:permutation_util", + "//xla:shape_util", + "//xla:status_macros", + "//xla:util", + "//xla:window_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/evaluator:hlo_evaluator", + "//xla/hlo/ir:hlo", + "//xla/hlo/ir:hlo_instruction_utils", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/utils:hlo_sharding_util", + "//xla/service:gather_scatter_utils", + "//xla/service:hlo_cost_analysis", + "//xla/service:hlo_creation_utils", + "//xla/service:hlo_module_config", + "//xla/service:host_memory_offload_annotations_hdr", + "//xla/service:host_offload_utils", + "//xla/service:pattern_matcher", + "//xla/service:shape_inference", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/numeric:bits", + "@com_google_absl//absl/status", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:status", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "algebraic_simplifier_test", + srcs = ["algebraic_simplifier_test.cc"], + deps = [ + ":algebraic_simplifier", + ":hlo_constant_folding", + "//xla:comparison_util", + "//xla:literal", + "//xla:literal_util", + "//xla:shape_util", + "//xla:test", + "//xla:util", + "//xla:window_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:hlo_creation_utils", + "//xla/service:host_memory_offload_annotations_hdr", + "//xla/service:layout_assignment", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/service:shape_inference", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/log", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "simplify_fp_conversions", + srcs = ["simplify_fp_conversions.cc"], + hdrs = ["simplify_fp_conversions.h"], + deps = [ + "//xla:comparison_util", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "simplify_fp_conversions_test", + srcs = ["simplify_fp_conversions_test.cc"], + deps = [ + ":simplify_fp_conversions", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:status_matchers", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "batch_dot_simplification", + srcs = ["batch_dot_simplification.cc"], + hdrs = ["batch_dot_simplification.h"], + deps = [ + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "batch_dot_simplification_test", + srcs = ["batch_dot_simplification_test.cc"], + deps = [ + ":batch_dot_simplification", + "//xla:test", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "convolution_group_converter", + srcs = ["convolution_group_converter.cc"], + hdrs = ["convolution_group_converter.h"], + deps = [ + "//xla:literal", + "//xla:literal_util", + "//xla:shape_util", + "//xla:status_macros", + "//xla:types", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:status", + ], +) + +xla_cc_test( + name = "convolution_group_converter_test", + size = "small", + srcs = ["convolution_group_converter_test.cc"], + deps = [ + ":convolution_group_converter", + "//xla:test", + "//xla:types", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "dot_dimension_merger", + srcs = ["dot_dimension_merger.cc"], + hdrs = ["dot_dimension_merger.h"], + deps = [ + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "dot_dimension_merger_test", + srcs = ["dot_dimension_merger_test.cc"], + deps = [ + ":dot_dimension_merger", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "dot_merger", + srcs = ["dot_merger.cc"], + hdrs = ["dot_merger.h"], + deps = [ + "//xla:protobuf_util", + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:shape_inference", + "//xla/service/graphcycles", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "dot_merger_test", + srcs = ["dot_merger_test.cc"], + deps = [ + ":algebraic_simplifier", + ":dot_merger", + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "convert_mover", + srcs = ["convert_mover.cc"], + hdrs = ["convert_mover.h"], + deps = [ + "//xla:literal", + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "convert_mover_test", + srcs = ["convert_mover_test.cc"], + deps = [ + ":convert_mover", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "tuple_simplifier", + srcs = ["tuple_simplifier.cc"], + hdrs = ["tuple_simplifier.h"], + deps = [ + "//xla:shape_util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "tuple_simplifier_test", + srcs = ["tuple_simplifier_test.cc"], + deps = [ + ":tuple_simplifier", + "//xla:shape_util", + "//xla:test", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "dynamic_dimension_simplifier", + srcs = ["dynamic_dimension_simplifier.cc"], + hdrs = ["dynamic_dimension_simplifier.h"], + deps = [ + "//xla:status_macros", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + ], +) + +xla_cc_test( + name = "dynamic_dimension_simplifier_test", + srcs = ["dynamic_dimension_simplifier_test.cc"], + deps = [ + ":dynamic_dimension_simplifier", + "//xla:literal", + "//xla:shape_util", + "//xla:test", + "//xla:types", + "//xla:window_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/pass:hlo_pass_pipeline", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:hlo_creation_utils", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/service:shape_inference", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:test_main", # fixdeps: keep + ], +) + +cc_library( + name = "reshape_mover", + srcs = ["reshape_mover.cc"], + hdrs = ["reshape_mover.h"], + deps = [ + "//xla:permutation_util", + "//xla:shape_util", + "//xla:status_macros", + "//xla:util", + "//xla/hlo/pass:hlo_pass", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "reshape_mover_test", + srcs = ["reshape_mover_test.cc"], + deps = [ + ":algebraic_simplifier", + ":reshape_mover", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:hlo_verifier", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/status", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "hlo_dce", + srcs = ["hlo_dce.cc"], + hdrs = ["hlo_dce.h"], + deps = [ + "//xla:shape_util", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "hlo_dce_test", + srcs = ["hlo_dce_test.cc"], + deps = [ + ":hlo_dce", + "//xla:literal_util", + "//xla:shape_util", + "//xla:types", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/tests:literal_test_util", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "hlo_rematerialization_test_utils", + testonly = 1, + hdrs = ["hlo_rematerialization_test_utils.h"], + deps = [ + "//xla:literal_util", + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@tsl//tsl/platform:test_main", + ], +) + +xla_cc_test( + name = "hlo_rematerialization_test_utils_test", + srcs = ["hlo_rematerialization_test_utils_test.cc"], + deps = [ + ":hlo_rematerialization_test_utils", + "//xla/hlo/ir:hlo", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "hlo_rematerialization", + srcs = ["hlo_rematerialization.cc"], + hdrs = ["hlo_rematerialization.h"], + deps = [ + ":hlo_dce", + "//xla:shape_util", + "//xla:status_macros", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/analysis:hlo_dataflow_analysis", + "//xla/hlo/analysis:tuple_points_to_analysis", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/utils:hlo_query", + "//xla/service:call_graph", + "//xla/service:hlo_cost_analysis", + "//xla/service:logical_buffer", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/functional:function_ref", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:numbers", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "hlo_rematerialization_test", + srcs = ["hlo_rematerialization_test.cc"], + deps = [ + ":hlo_memory_scheduler", + ":hlo_rematerialization", + ":hlo_rematerialization_test_utils", + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/utils:hlo_matchers", + "//xla/service:hlo_cost_analysis", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "hlo_constant_folding", + srcs = ["hlo_constant_folding.cc"], + hdrs = ["hlo_constant_folding.h"], + deps = [ + "//xla:literal", + "//xla:shape_util", + "//xla/hlo/evaluator:hlo_evaluator", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:slow_operation_alarm", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/time", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "hlo_constant_folding_test", + srcs = ["hlo_constant_folding_test.cc"], + deps = [ + ":hlo_constant_folding", + "//xla:literal", + "//xla:literal_util", + "//xla:permutation_util", + "//xla:shape_util", + "//xla:test", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +# copybara:uncomment_begin(google-only) +# xla_cc_test( +# name = "hlo_constant_folding_peak_heap_test", +# srcs = ["hlo_constant_folding_peak_heap_test.cc"], +# deps = [ +# ":hlo_constant_folding", +# "@com_google_googletest//:gtest", +# "@com_google_absl//absl/strings:str_format", +# "//xla:test", +# "//xla/hlo/testlib:hlo_hardware_independent_test_base", +# "@tsl//tsl/platform:statusor", +# "@tsl//tsl/platform:test_main", +# ], +# ) +# copybara:uncomment_end + +cc_library( + name = "hlo_element_type_converter", + srcs = ["hlo_element_type_converter.cc"], + hdrs = ["hlo_element_type_converter.h"], + deps = [ + "//xla:literal", + "//xla:shape_util", + "//xla:types", + "//xla:xla_data_proto_cc", + "//xla/hlo/evaluator:hlo_evaluator", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/utils:hlo_query", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "hlo_element_type_converter_test", + srcs = ["hlo_element_type_converter_test.cc"], + deps = [ + ":hlo_element_type_converter", + "//xla:xla_data_proto_cc", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "conditional_canonicalizer", + srcs = ["conditional_canonicalizer.cc"], + hdrs = ["conditional_canonicalizer.h"], + deps = [ + "//xla:status_macros", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + ], +) + +xla_cc_test( + name = "conditional_canonicalizer_test", + srcs = ["conditional_canonicalizer_test.cc"], + deps = [ + ":conditional_canonicalizer", + "//xla:shape_util", + "//xla:types", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "//xla/tests:literal_test_util", + "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "zero_sized_hlo_elimination", + srcs = ["zero_sized_hlo_elimination.cc"], + hdrs = ["zero_sized_hlo_elimination.h"], + deps = [ + "//xla:literal", + "//xla:shape_util", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "zero_sized_hlo_elimination_test", + srcs = ["zero_sized_hlo_elimination_test.cc"], + deps = [ + ":zero_sized_hlo_elimination", + "//xla:literal_util", + "//xla:shape_util", + "//xla:test", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@tsl//tsl/platform:logging", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "sort_simplifier", + srcs = ["sort_simplifier.cc"], + hdrs = ["sort_simplifier.h"], + deps = [ + "//xla:shape_util", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "sort_simplifier_test", + srcs = ["sort_simplifier_test.cc"], + deps = [ + ":sort_simplifier", + "//xla:test", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/tsl/lib/core:status_test_util", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "root_instruction_sinker", + srcs = ["root_instruction_sinker.cc"], + hdrs = ["root_instruction_sinker.h"], + deps = [ + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:tuple_util", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + ], +) + +xla_cc_test( + name = "root_instruction_sinker_test", + srcs = ["root_instruction_sinker_test.cc"], + deps = [ + ":root_instruction_sinker", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "host_memory_transfer_asyncifier", + srcs = ["host_memory_transfer_asyncifier.cc"], + hdrs = ["host_memory_transfer_asyncifier.h"], + deps = [ + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "host_memory_transfer_asyncifier_test", + srcs = ["host_memory_transfer_asyncifier_test.cc"], + deps = [ + ":host_memory_transfer_asyncifier", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "fusion_constant_sinking", + srcs = ["fusion_constant_sinking.cc"], + hdrs = ["fusion_constant_sinking.h"], + deps = [ + ":hlo_dce", + "//xla:shape_util", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "fusion_constant_sinking_test", + srcs = ["fusion_constant_sinking_test.cc"], + deps = [ + ":fusion_constant_sinking", + "//xla:test", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "optimize_input_output_buffer_alias", + srcs = ["optimize_input_output_buffer_alias.cc"], + hdrs = ["optimize_input_output_buffer_alias.h"], + deps = [ + "//xla:shape_util", + "//xla:status_macros", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "optimize_input_output_buffer_alias_test", + srcs = ["optimize_input_output_buffer_alias_test.cc"], + deps = [ + ":optimize_input_output_buffer_alias", + "//xla:shape_util", + "//xla:test", + "//xla:test_helpers", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_absl//absl/status", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:test", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "ar_crs_combiner", + srcs = ["ar_crs_combiner.cc"], + hdrs = ["ar_crs_combiner.h"], + deps = [ + "//xla:literal", + "//xla:literal_util", + "//xla:shape_util", + "//xla:status_macros", + "//xla:xla_data_proto_cc", + "//xla/hlo/analysis:hlo_replication_analysis", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/hlo/utils:hlo_query", + "//xla/service:call_graph", + "//xla/service:pattern_matcher", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:status", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "ar_crs_combiner_test", + srcs = ["ar_crs_combiner_test.cc"], + deps = [ + ":ar_crs_combiner", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "slice_sinker", + srcs = ["slice_sinker.cc"], + hdrs = ["slice_sinker.h"], + deps = [ + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "slice_sinker_test", + srcs = ["slice_sinker_test.cc"], + deps = [ + ":hlo_dce", + ":slice_sinker", + "//xla:literal_util", + "//xla:shape_util", + "//xla:types", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "result_caster", + srcs = ["result_caster.cc"], + hdrs = ["result_caster.h"], + deps = [ + "//xla:shape_util", + "//xla:util", + "//xla/hlo/ir:hlo", + "//xla/hlo/transforms:op_expander_pass", + "//xla/service:shape_inference", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + ], +) + +xla_cc_test( + name = "result_caster_test", + srcs = ["result_caster_test.cc"], + deps = [ + ":result_caster", + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "convert_operand_folding", + srcs = ["convert_operand_folder.cc"], + hdrs = ["convert_operand_folder.h"], + deps = [ + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/transforms:op_expander_pass", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:errors", + ], +) + +xla_cc_test( + name = "convert_operand_folding_test", + srcs = ["convert_operand_folder_test.cc"], + deps = [ + ":convert_operand_folding", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "reduce_window_rewriter", + srcs = ["reduce_window_rewriter.cc"], + hdrs = ["reduce_window_rewriter.h"], + deps = [ + "//xla:shape_util", + "//xla:status_macros", + "//xla:util", + "//xla:window_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "reduce_window_rewriter_test", + srcs = ["reduce_window_rewriter_test.cc"], + deps = [ + ":reduce_window_rewriter", + "//xla:test", + "//xla:xla_data_proto_cc", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "gather_simplifier", + srcs = ["gather_simplifier.cc"], + hdrs = ["gather_simplifier.h"], + deps = [ + "//xla:literal_util", + "//xla:permutation_util", + "//xla:shape_util", + "//xla/hlo/ir:hlo", + "//xla/hlo/transforms:op_expander_pass", + "//xla/service:gather_scatter_utils", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "gather_simplifier_test", + srcs = ["gather_simplifier_test.cc"], + deps = [ + ":gather_simplifier", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@tsl//tsl/platform:test_main", + ], +) + +cc_library( + name = "instruction_hoister", + srcs = ["instruction_hoister.cc"], + hdrs = ["instruction_hoister.h"], + deps = [ + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@tsl//tsl/platform:status", + ], +) + +cc_library( + name = "sub_byte_normalization", + srcs = ["sub_byte_normalization.cc"], + hdrs = ["sub_byte_normalization.h"], + deps = [ + "//xla:shape_layout", + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:status", + ], +) + +cc_library( + name = "tree_reduction_rewriter", + srcs = ["tree_reduction_rewriter.cc"], + hdrs = ["tree_reduction_rewriter.h"], + deps = [ + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/builder:padding", + "//xla/hlo/ir:hlo", + "//xla/hlo/pass:hlo_pass", + "//xla/service:shape_inference", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@tsl//tsl/platform:errors", + "@tsl//tsl/platform:status", + ], +) diff --git a/xla/python/BUILD b/xla/python/BUILD index 6283dee12f3e49..4b024b2c8c3df0 100644 --- a/xla/python/BUILD +++ b/xla/python/BUILD @@ -1140,9 +1140,9 @@ cc_library( "//xla/hlo/ir:hlo_module_group", "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:flatten_call_graph", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:flatten_call_graph", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/pjrt:compile_options_proto_cc", "//xla/pjrt:exceptions", "//xla/pjrt:pjrt_executable", diff --git a/xla/service/BUILD b/xla/service/BUILD index b70e65c0e2e600..7d7643f7593541 100644 --- a/xla/service/BUILD +++ b/xla/service/BUILD @@ -214,8 +214,8 @@ xla_cc_test( cc_library( name = "all_reduce_folder", hdrs = ["all_reduce_folder.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:all_reduce_folder instead.", - deps = ["//xla/hlo/transforms:all_reduce_folder"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:all_reduce_folder instead.", + deps = ["//xla/hlo/transforms/simplifiers:all_reduce_folder"], ) cc_library( @@ -231,22 +231,22 @@ cc_library( cc_library( name = "broadcast_canonicalizer", hdrs = ["broadcast_canonicalizer.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:broadcast_canonicalizer instead.", - deps = ["//xla/hlo/transforms:broadcast_canonicalizer"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:broadcast_canonicalizer instead.", + deps = ["//xla/hlo/transforms/simplifiers:broadcast_canonicalizer"], ) cc_library( name = "bfloat16_conversion_folding", hdrs = ["bfloat16_conversion_folding.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:bfloat16_conversion_folding instead.", - deps = ["//xla/hlo/transforms:bfloat16_conversion_folding"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:bfloat16_conversion_folding instead.", + deps = ["//xla/hlo/transforms/simplifiers:bfloat16_conversion_folding"], ) cc_library( name = "float_normalization", hdrs = ["float_normalization.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:float_normalization instead.", - deps = ["//xla/hlo/transforms:float_normalization"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:float_normalization instead.", + deps = ["//xla/hlo/transforms/simplifiers:float_normalization"], ) cc_library( @@ -381,7 +381,7 @@ cc_library( "//xla/hlo/ir:hlo_instruction_utils", "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -610,8 +610,8 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:hlo_constant_splitter", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_constant_splitter", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", @@ -915,8 +915,8 @@ xla_cc_test( cc_library( name = "flatten_call_graph", hdrs = ["flatten_call_graph.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:flatten_call_graph instead.", - deps = ["//xla/hlo/transforms:flatten_call_graph"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:flatten_call_graph instead.", + deps = ["//xla/hlo/transforms/simplifiers:flatten_call_graph"], ) cc_library( @@ -932,7 +932,7 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service/spmd/shardy:constants", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -973,8 +973,8 @@ xla_cc_test( cc_library( name = "hlo_computation_deduplicator", hdrs = ["hlo_computation_deduplicator.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:hlo_computation_deduplicator instead.", - deps = ["//xla/hlo/transforms:hlo_computation_deduplicator"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:hlo_computation_deduplicator instead.", + deps = ["//xla/hlo/transforms/simplifiers:hlo_computation_deduplicator"], ) cc_library( @@ -1723,9 +1723,9 @@ xla_cc_test( "//xla/hlo/analysis:hlo_ordering", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:flatten_call_graph", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:flatten_call_graph", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", "//xla/service/memory_space_assignment", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", @@ -1843,8 +1843,8 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/analysis:hlo_ordering", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", @@ -1876,9 +1876,9 @@ xla_cc_test( cc_library( name = "hlo_memory_scheduler", hdrs = ["hlo_memory_scheduler.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:hlo_memory_scheduler instead.", + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:hlo_memory_scheduler instead.", local_defines = if_cuda_is_configured(["GOOGLE_CUDA=1"]), - deps = ["//xla/hlo/transforms:hlo_memory_scheduler"], + deps = ["//xla/hlo/transforms/simplifiers:hlo_memory_scheduler"], ) cc_library( @@ -1950,7 +1950,7 @@ cc_library( "//xla/hlo/analysis:hlo_reachability", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status:statusor", @@ -2310,15 +2310,15 @@ cc_library( name = "algebraic_simplifier", hdrs = ["algebraic_simplifier.h"], copts = tsl_copts(), - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:algebraic_simplifier instead.", - deps = ["//xla/hlo/transforms:algebraic_simplifier"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:algebraic_simplifier instead.", + deps = ["//xla/hlo/transforms/simplifiers:algebraic_simplifier"], ) cc_library( name = "tree_reduction_rewriter", hdrs = ["tree_reduction_rewriter.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:tree_reduction_rewriter instead.", - deps = ["//xla/hlo/transforms:tree_reduction_rewriter"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:tree_reduction_rewriter instead.", + deps = ["//xla/hlo/transforms/simplifiers:tree_reduction_rewriter"], ) xla_test( @@ -2335,8 +2335,8 @@ xla_test( cc_library( name = "simplify_fp_conversions", hdrs = ["simplify_fp_conversions.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:simplify_fp_conversions instead.", - deps = ["//xla/hlo/transforms:simplify_fp_conversions"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:simplify_fp_conversions instead.", + deps = ["//xla/hlo/transforms/simplifiers:simplify_fp_conversions"], ) cc_library( @@ -2570,8 +2570,8 @@ xla_cc_test( cc_library( name = "batch_dot_simplification", hdrs = ["batch_dot_simplification.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:batch_dot_simplification instead.", - deps = ["//xla/hlo/transforms:batch_dot_simplification"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:batch_dot_simplification instead.", + deps = ["//xla/hlo/transforms/simplifiers:batch_dot_simplification"], ) xla_cc_test( @@ -2648,8 +2648,8 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:inlined_vector", @@ -2685,8 +2685,8 @@ xla_cc_test( cc_library( name = "convolution_group_converter", hdrs = ["convolution_group_converter.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:convolution_group_converter instead.", - deps = ["//xla/hlo/transforms:convolution_group_converter"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:convolution_group_converter instead.", + deps = ["//xla/hlo/transforms/simplifiers:convolution_group_converter"], ) cc_library( @@ -2757,7 +2757,7 @@ cc_library( "//xla/hlo/analysis:hlo_dataflow_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/status:statusor", @@ -2800,7 +2800,7 @@ cc_library( "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -2853,8 +2853,8 @@ cc_library( "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:flatten_call_graph", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:flatten_call_graph", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm", "@com_google_absl//absl/algorithm:container", @@ -2914,7 +2914,7 @@ cc_library( "//xla/hlo/analysis:while_loop_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -2941,8 +2941,8 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", @@ -2977,22 +2977,22 @@ cc_library( cc_library( name = "dot_dimension_merger", hdrs = ["dot_dimension_merger.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:dot_dimension_merger instead.", - deps = ["//xla/hlo/transforms:dot_dimension_merger"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:dot_dimension_merger instead.", + deps = ["//xla/hlo/transforms/simplifiers:dot_dimension_merger"], ) cc_library( name = "dot_merger", hdrs = ["dot_merger.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:dot_merger instead.", - deps = ["//xla/hlo/transforms:dot_merger"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:dot_merger instead.", + deps = ["//xla/hlo/transforms/simplifiers:dot_merger"], ) cc_library( name = "convert_mover", hdrs = ["convert_mover.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:convert_mover instead.", - deps = ["//xla/hlo/transforms:convert_mover"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:convert_mover instead.", + deps = ["//xla/hlo/transforms/simplifiers:convert_mover"], ) cc_library( @@ -3050,15 +3050,15 @@ xla_cc_test( cc_library( name = "tuple_simplifier", hdrs = ["tuple_simplifier.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:tuple_simplifier instead.", - deps = ["//xla/hlo/transforms:tuple_simplifier"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:tuple_simplifier instead.", + deps = ["//xla/hlo/transforms/simplifiers:tuple_simplifier"], ) cc_library( name = "reshape_mover", hdrs = ["reshape_mover.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:reshape_mover instead.", - deps = ["//xla/hlo/transforms:reshape_mover"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:reshape_mover instead.", + deps = ["//xla/hlo/transforms/simplifiers:reshape_mover"], ) cc_library( @@ -3133,8 +3133,8 @@ cc_library( cc_library( name = "dynamic_dimension_simplifier", hdrs = ["dynamic_dimension_simplifier.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:dynamic_dimension_simplifier instead.", - deps = ["//xla/hlo/transforms:dynamic_dimension_simplifier"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:dynamic_dimension_simplifier instead.", + deps = ["//xla/hlo/transforms/simplifiers:dynamic_dimension_simplifier"], ) cc_library( @@ -3159,7 +3159,7 @@ cc_library( "//xla/hlo/builder:xla_builder", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/tsl/lib/monitoring:gauge", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", @@ -3197,10 +3197,10 @@ xla_test( "//xla/hlo/builder:xla_builder", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:dynamic_dimension_simplifier", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:dynamic_dimension_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/utils:hlo_matchers", "//xla/tests:client_library_test_base", "//xla/tests:hlo_test_base", @@ -3501,7 +3501,7 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", @@ -3723,8 +3723,8 @@ cc_library( "//xla/hlo/analysis:tuple_points_to_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -3768,8 +3768,8 @@ cc_library( "//xla/hlo/analysis:hlo_reachability", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/utils:hlo_query", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", @@ -3871,8 +3871,8 @@ cc_library( cc_library( name = "hlo_dce", hdrs = ["hlo_dce.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:hlo_dce instead.", - deps = ["//xla/hlo/transforms:hlo_dce"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:hlo_dce instead.", + deps = ["//xla/hlo/transforms/simplifiers:hlo_dce"], ) cc_library( @@ -3887,8 +3887,8 @@ cc_library( "//xla/hlo/analysis:hlo_liveness_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@tsl//tsl/platform:errors", @@ -3992,16 +3992,16 @@ xla_cc_test( cc_library( name = "hlo_rematerialization", hdrs = ["hlo_rematerialization.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:hlo_rematerialization instead.", - deps = ["//xla/hlo/transforms:hlo_rematerialization"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:hlo_rematerialization instead.", + deps = ["//xla/hlo/transforms/simplifiers:hlo_rematerialization"], ) cc_library( name = "hlo_rematerialization_test_utils", testonly = 1, hdrs = ["hlo_rematerialization_test_utils.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:hlo_rematerialization_test_utils instead.", - deps = ["//xla/hlo/transforms:hlo_rematerialization_test_utils"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:hlo_rematerialization_test_utils instead.", + deps = ["//xla/hlo/transforms/simplifiers:hlo_rematerialization_test_utils"], ) xla_cc_test( @@ -4039,7 +4039,7 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", @@ -4130,7 +4130,7 @@ xla_cc_test( cc_library( name = "hlo_constant_folding", hdrs = ["hlo_constant_folding.h"], - deps = ["//xla/hlo/transforms:hlo_constant_folding"], + deps = ["//xla/hlo/transforms/simplifiers:hlo_constant_folding"], ) cc_library( @@ -4216,15 +4216,15 @@ xla_cc_test( cc_library( name = "hlo_element_type_converter", hdrs = ["hlo_element_type_converter.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:hlo_element_type_converter instead.", - deps = ["//xla/hlo/transforms:hlo_element_type_converter"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:hlo_element_type_converter instead.", + deps = ["//xla/hlo/transforms/simplifiers:hlo_element_type_converter"], ) cc_library( name = "conditional_canonicalizer", hdrs = ["conditional_canonicalizer.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:conditional_canonicalizer instead.", - deps = ["//xla/hlo/transforms:conditional_canonicalizer"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:conditional_canonicalizer instead.", + deps = ["//xla/hlo/transforms/simplifiers:conditional_canonicalizer"], ) cc_library( @@ -4500,8 +4500,8 @@ xla_cc_test( cc_library( name = "zero_sized_hlo_elimination", hdrs = ["zero_sized_hlo_elimination.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:zero_sized_hlo_elimination instead.", - deps = ["//xla/hlo/transforms:zero_sized_hlo_elimination"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:zero_sized_hlo_elimination instead.", + deps = ["//xla/hlo/transforms/simplifiers:zero_sized_hlo_elimination"], ) cc_library( @@ -4661,8 +4661,8 @@ cc_library( cc_library( name = "sort_simplifier", hdrs = ["sort_simplifier.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:sort_simplifier instead.", - deps = ["//xla/hlo/transforms:sort_simplifier"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:sort_simplifier instead.", + deps = ["//xla/hlo/transforms/simplifiers:sort_simplifier"], ) cc_library( @@ -4713,8 +4713,8 @@ xla_cc_test( cc_library( name = "root_instruction_sinker", hdrs = ["root_instruction_sinker.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:root_instruction_sinker instead.", - deps = ["//xla/hlo/transforms:root_instruction_sinker"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:root_instruction_sinker instead.", + deps = ["//xla/hlo/transforms/simplifiers:root_instruction_sinker"], ) cc_library( @@ -4735,8 +4735,8 @@ cc_library( cc_library( name = "host_memory_transfer_asyncifier", hdrs = ["host_memory_transfer_asyncifier.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:host_memory_transfer_asyncifier instead.", - deps = ["//xla/hlo/transforms:host_memory_transfer_asyncifier"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:host_memory_transfer_asyncifier instead.", + deps = ["//xla/hlo/transforms/simplifiers:host_memory_transfer_asyncifier"], ) cc_library( @@ -4918,8 +4918,8 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -4962,7 +4962,7 @@ cc_library( "//xla/hlo/analysis:while_loop_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", @@ -5035,8 +5035,8 @@ xla_cc_test( cc_library( name = "fusion_constant_sinking", hdrs = ["fusion_constant_sinking.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:fusion_constant_sinking instead.", - deps = ["//xla/hlo/transforms:fusion_constant_sinking"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:fusion_constant_sinking instead.", + deps = ["//xla/hlo/transforms/simplifiers:fusion_constant_sinking"], ) cc_library( @@ -5112,7 +5112,7 @@ xla_cc_test( deps = [ ":while_loop_fusible_sinking", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:flatten_call_graph", + "//xla/hlo/transforms/simplifiers:flatten_call_graph", "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "@com_google_absl//absl/log:check", @@ -5188,15 +5188,15 @@ cc_library( cc_library( name = "optimize_input_output_buffer_alias", hdrs = ["optimize_input_output_buffer_alias.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:optimize_input_output_buffer_alias instead.", - deps = ["//xla/hlo/transforms:optimize_input_output_buffer_alias"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:optimize_input_output_buffer_alias instead.", + deps = ["//xla/hlo/transforms/simplifiers:optimize_input_output_buffer_alias"], ) cc_library( name = "ar_crs_combiner", hdrs = ["ar_crs_combiner.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:ar_crs_combiner instead.", - deps = ["//xla/hlo/transforms:ar_crs_combiner"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:ar_crs_combiner instead.", + deps = ["//xla/hlo/transforms/simplifiers:ar_crs_combiner"], ) cc_library( @@ -5283,7 +5283,7 @@ cc_library( name = "slice_sinker", hdrs = ["slice_sinker.h"], deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:slice_sinker instead.", - deps = ["//xla/hlo/transforms:slice_sinker"], + deps = ["//xla/hlo/transforms/simplifiers:slice_sinker"], ) cc_library( @@ -5515,8 +5515,8 @@ xla_cc_test( ":pattern_matcher_gmock", ":topk_rewriter", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "//xla/tests:literal_test_util", @@ -5537,8 +5537,8 @@ cc_library( cc_library( name = "result_caster", hdrs = ["result_caster.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:result_caster instead.", - deps = ["//xla/hlo/transforms:result_caster"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:result_caster instead.", + deps = ["//xla/hlo/transforms/simplifiers:result_caster"], ) cc_library( @@ -5556,8 +5556,8 @@ cc_library( cc_library( name = "convert_operand_folding", hdrs = ["convert_operand_folding.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:convert_operand_folding instead.", - deps = ["//xla/hlo/transforms:convert_operand_folding"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:convert_operand_folding instead.", + deps = ["//xla/hlo/transforms/simplifiers:convert_operand_folding"], ) cc_library( @@ -5798,8 +5798,8 @@ cc_library( cc_library( name = "instruction_hoister", hdrs = ["instruction_hoister.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:instruction_hoister instead.", - deps = ["//xla/hlo/transforms:instruction_hoister"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:instruction_hoister instead.", + deps = ["//xla/hlo/transforms/simplifiers:instruction_hoister"], ) cc_library( @@ -5913,7 +5913,7 @@ cc_library( name = "gather_simplifier", hdrs = ["gather_simplifier.h"], deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:gather_simplifier instead.", - deps = ["//xla/hlo/transforms:gather_simplifier"], + deps = ["//xla/hlo/transforms/simplifiers:gather_simplifier"], ) cc_library( @@ -5939,8 +5939,8 @@ cc_library( cc_library( name = "reduce_window_rewriter", hdrs = ["reduce_window_rewriter.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:reduce_window_rewriter instead.", - deps = ["//xla/hlo/transforms:reduce_window_rewriter"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:reduce_window_rewriter instead.", + deps = ["//xla/hlo/transforms/simplifiers:reduce_window_rewriter"], ) cc_library( @@ -5964,8 +5964,8 @@ cc_library( cc_library( name = "sub_byte_normalization", hdrs = ["sub_byte_normalization.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:sub_byte_normalization instead.", - deps = ["//xla/hlo/transforms:sub_byte_normalization"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/simplifiers:sub_byte_normalization instead.", + deps = ["//xla/hlo/transforms/simplifiers:sub_byte_normalization"], ) cc_library( @@ -6320,8 +6320,8 @@ cc_library( ":while_util", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:flatten_call_graph", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:flatten_call_graph", + "//xla/hlo/transforms/simplifiers:hlo_dce", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log:check", diff --git a/xla/service/cpu/BUILD b/xla/service/cpu/BUILD index 8f35ed919dd78d..1e7707f0df9c0c 100644 --- a/xla/service/cpu/BUILD +++ b/xla/service/cpu/BUILD @@ -255,45 +255,45 @@ cc_library( "//xla/hlo/ir:hlo_module_group", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:batch_dot_simplification", "//xla/hlo/transforms:bitcast_dtypes_expander", - "//xla/hlo/transforms:broadcast_canonicalizer", "//xla/hlo/transforms:cholesky_expander", "//xla/hlo/transforms:comparison_expander", - "//xla/hlo/transforms:conditional_canonicalizer", - "//xla/hlo/transforms:convolution_group_converter", "//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:hlo_constant_folding", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:hlo_memory_scheduler", "//xla/hlo/transforms:literal_canonicalizer", "//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:reduce_decomposer", - "//xla/hlo/transforms:reduce_window_rewriter", "//xla/hlo/transforms:reshape_decomposer", - "//xla/hlo/transforms:reshape_mover", - "//xla/hlo/transforms:result_caster", "//xla/hlo/transforms:rng_bit_generator_expander", "//xla/hlo/transforms:rng_expander", - "//xla/hlo/transforms:simplify_fp_conversions", - "//xla/hlo/transforms:slice_sinker", - "//xla/hlo/transforms:sort_simplifier", "//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/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:batch_dot_simplification", + "//xla/hlo/transforms/simplifiers:broadcast_canonicalizer", + "//xla/hlo/transforms/simplifiers:conditional_canonicalizer", + "//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:hlo_constant_folding", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:optimize_input_output_buffer_alias", + "//xla/hlo/transforms/simplifiers:reduce_window_rewriter", + "//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/translate/hlo_to_mhlo:hlo_to_mlir_hlo", "//xla/hlo/translate/hlo_to_mhlo:hlo_utils", "//xla/mlir_hlo", @@ -700,7 +700,7 @@ xla_cc_test( "//xla/hlo/analysis:hlo_ordering", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", "//xla/service:buffer_assignment", "//xla/service:buffer_value", "//xla/service:hlo_module_config", diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD index c0552ed4936bda..3bf55230507d43 100644 --- a/xla/service/gpu/BUILD +++ b/xla/service/gpu/BUILD @@ -1323,7 +1323,7 @@ cc_library( "//xla:xla_proto_cc", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:cpu_gpu_shape_verifier", "//xla/service:hlo_cost_analysis", "//xla/service:hlo_cse", @@ -1349,7 +1349,7 @@ cc_library( "//xla/hlo/analysis:hlo_dataflow_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:copy_insertion", "//xla/service:cpu_gpu_shape_verifier", "//xla/service:hlo_verifier", @@ -1441,53 +1441,53 @@ cc_library( "//xla/hlo/transforms/collectives:collective_quantizer", "//xla/hlo/transforms/collectives:collectives_schedule_linearizer", "//xla/hlo/transforms/collectives:convert_async_collectives_to_sync", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:all_reduce_folder", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:all_reduce_folder", + "//xla/hlo/transforms/simplifiers:broadcast_canonicalizer", + "//xla/hlo/transforms/simplifiers:conditional_canonicalizer", + "//xla/hlo/transforms/simplifiers:convert_mover", + "//xla/hlo/transforms/simplifiers:dot_merger", + "//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_computation_deduplicator", + "//xla/hlo/transforms/simplifiers:hlo_constant_folding", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_rematerialization", + "//xla/hlo/transforms/simplifiers:host_memory_transfer_asyncifier", + "//xla/hlo/transforms/simplifiers:optimize_input_output_buffer_alias", + "//xla/hlo/transforms/simplifiers:reduce_window_rewriter", + "//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:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:zero_sized_hlo_elimination", "//xla/hlo/transforms:bitcast_dtypes_expander", - "//xla/hlo/transforms:broadcast_canonicalizer", "//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_pred_expander", "//xla/hlo/transforms:dot_decomposer", - "//xla/hlo/transforms:dot_merger", - "//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_computation_deduplicator", - "//xla/hlo/transforms:hlo_constant_folding", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:hlo_rematerialization", - "//xla/hlo/transforms:host_memory_transfer_asyncifier", "//xla/hlo/transforms:host_offload_legalize", "//xla/hlo/transforms:host_offloader", "//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:reduce_window_rewriter", "//xla/hlo/transforms:reshape_decomposer", - "//xla/hlo/transforms:reshape_mover", - "//xla/hlo/transforms:result_caster", "//xla/hlo/transforms:rng_bit_generator_expander", "//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:tuple_simplifier", "//xla/hlo/transforms:while_loop_trip_count_annotator", - "//xla/hlo/transforms:zero_sized_hlo_elimination", "//xla/hlo/translate/hlo_to_mhlo:hlo_utils", "//xla/hlo/translate/mhlo_to_hlo:location_exporter", "//xla/hlo/utils:hlo_query", @@ -1698,8 +1698,8 @@ xla_test( "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:hlo_memory_scheduler", - "//xla/hlo/transforms:hlo_rematerialization", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:hlo_rematerialization", "//xla/hlo/utils:hlo_matchers", "//xla/service:buffer_value", "//xla/service:hlo_cost_analysis", @@ -1783,14 +1783,14 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:convert_mover", - "//xla/hlo/transforms:dot_dimension_merger", - "//xla/hlo/transforms:float_normalization", - "//xla/hlo/transforms:hlo_constant_folding", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:reshape_mover", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:convert_mover", + "//xla/hlo/transforms/simplifiers:dot_dimension_merger", + "//xla/hlo/transforms/simplifiers:float_normalization", + "//xla/hlo/transforms/simplifiers:hlo_constant_folding", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:reshape_mover", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/pjrt/distributed:key_value_store_interface", "//xla/service:call_inliner", "//xla/service:dump", @@ -2021,13 +2021,13 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:convert_mover", - "//xla/hlo/transforms:dot_dimension_merger", - "//xla/hlo/transforms:float_normalization", - "//xla/hlo/transforms:hlo_constant_folding", - "//xla/hlo/transforms:reshape_mover", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:convert_mover", + "//xla/hlo/transforms/simplifiers:dot_dimension_merger", + "//xla/hlo/transforms/simplifiers:float_normalization", + "//xla/hlo/transforms/simplifiers:hlo_constant_folding", + "//xla/hlo/transforms/simplifiers:reshape_mover", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/pjrt/distributed:key_value_store_interface", "//xla/service:call_inliner", "//xla/service:float_support", @@ -2120,7 +2120,7 @@ cc_library( "//xla:util", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_memory_scheduler", + "//xla/hlo/transforms/simplifiers:hlo_memory_scheduler", "//xla/hlo/utils:hlo_query", "//xla/service:buffer_value", "//xla/service:collective_ops_utils", @@ -2233,13 +2233,13 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:hlo_constant_folding", - "//xla/hlo/transforms:hlo_constant_splitter", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:reshape_mover", - "//xla/hlo/transforms:sort_simplifier", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_constant_folding", + "//xla/hlo/transforms/simplifiers:hlo_constant_splitter", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:reshape_mover", + "//xla/hlo/transforms/simplifiers:sort_simplifier", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/service:conditional_simplifier", "//xla/service:gather_expander", "//xla/service:hlo_module_config", @@ -2270,7 +2270,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/service:hlo_module_config", "//xla/service/spmd/shardy:constants", "//xla/stream_executor:device_description", @@ -2671,7 +2671,7 @@ xla_cc_test( "//xla:test_helpers", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:float_normalization", + "//xla/hlo/transforms/simplifiers:float_normalization", "//xla/service:hlo_verifier", "//xla/stream_executor:device_description", "//xla/tests:hlo_test_base", @@ -3091,7 +3091,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/hlo/utils:hlo_query", "//xla/service:collective_pipeliner", "//xla/service:collective_utils", @@ -3260,7 +3260,7 @@ xla_cc_test( deps = [ ":flag_utils", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:collective_pipeliner", "//xla/service:hlo_module_config", "//xla/service:latency_hiding_scheduler", diff --git a/xla/service/gpu/autotuning/BUILD b/xla/service/gpu/autotuning/BUILD index 109bb634e22349..3ab69229d1c1f5 100644 --- a/xla/service/gpu/autotuning/BUILD +++ b/xla/service/gpu/autotuning/BUILD @@ -127,7 +127,7 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:float_normalization", + "//xla/hlo/transforms/simplifiers:float_normalization", "//xla/hlo/utils:hlo_query", "//xla/hlo/utils:hlo_traversal", "//xla/pjrt/distributed:key_value_store_interface", @@ -515,7 +515,7 @@ xla_test( "//xla:xla_data_proto_cc", "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", "//xla/service:platform_util", diff --git a/xla/service/gpu/fusions/triton/BUILD b/xla/service/gpu/fusions/triton/BUILD index 97b91379f63a5b..e120f6fe8686e1 100644 --- a/xla/service/gpu/fusions/triton/BUILD +++ b/xla/service/gpu/fusions/triton/BUILD @@ -766,7 +766,7 @@ cc_library( "//xla/hlo/pass:hlo_pass_pipeline", "//xla/hlo/testlib:filecheck", "//xla/hlo/testlib:verified_hlo_module", - "//xla/hlo/transforms:float_normalization", + "//xla/hlo/transforms/simplifiers:float_normalization", "//xla/hlo/utils:hlo_query", "//xla/service/gpu:backend_configs_cc", "//xla/service/gpu:gpu_device_info_for_tests", diff --git a/xla/service/gpu/transforms/BUILD b/xla/service/gpu/transforms/BUILD index 3270b6432a542a..f8f0e93a387f52 100644 --- a/xla/service/gpu/transforms/BUILD +++ b/xla/service/gpu/transforms/BUILD @@ -76,7 +76,7 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/service:hlo_creation_utils", "//xla/service:pattern_matcher", "//xla/service/gpu:matmul_utils", @@ -98,7 +98,7 @@ xla_cc_test( deps = [ ":algebraic_simplifier", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", "//xla/stream_executor:device_description", @@ -947,10 +947,10 @@ xla_test( "//xla/hlo/pass:hlo_pass_pipeline", "//xla/hlo/testlib:filecheck", "//xla/hlo/testlib:verified_hlo_module", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:convert_mover", - "//xla/hlo/transforms:hlo_constant_folding", - "//xla/hlo/transforms:reshape_mover", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:convert_mover", + "//xla/hlo/transforms/simplifiers:hlo_constant_folding", + "//xla/hlo/transforms/simplifiers:reshape_mover", "//xla/service:hlo_module_config", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", @@ -1030,9 +1030,9 @@ xla_test( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:hlo_dce", "//xla/hlo/transforms:reshape_decomposer", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:computation_layout", "//xla/service:hlo_cse", "//xla/service:hlo_module_config", @@ -1254,9 +1254,9 @@ xla_cc_test( "//xla:util", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:reshape_mover", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:reshape_mover", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/service:call_inliner", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", @@ -1563,7 +1563,7 @@ cc_library( "//xla/hlo/ir:hlo_instruction_utils", "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:flatten_call_graph", + "//xla/hlo/transforms/simplifiers:flatten_call_graph", "//xla/hlo/utils:hlo_query", "//xla/service:collective_ops_utils", "@com_google_absl//absl/algorithm:container", @@ -1591,7 +1591,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass_pipeline", "//xla/hlo/testlib:filecheck", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/utils:hlo_query", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", @@ -2089,7 +2089,7 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:sub_byte_normalization", + "//xla/hlo/transforms/simplifiers:sub_byte_normalization", "//xla/service:hlo_creation_utils", "//xla/service/gpu:gpu_fusible", "//xla/stream_executor:device_description", @@ -2122,7 +2122,7 @@ xla_test( "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:hlo_cost_analysis", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", @@ -2293,7 +2293,7 @@ cc_library( deps = [ "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/hlo/utils:hlo_query", "//xla/service:call_graph", "//xla/service:instruction_fusion", @@ -3106,7 +3106,7 @@ xla_cc_test( deps = [ ":topk_splitter", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:pattern_matcher", "//xla/service:topk_rewriter", "//xla/tests:hlo_test_base", @@ -3341,8 +3341,8 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:hlo_constant_folding", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_constant_folding", "//xla/hlo/utils:hlo_query", "//xla/service:hlo_creation_utils", "//xla/service:pattern_matcher", diff --git a/xla/service/memory_space_assignment/BUILD b/xla/service/memory_space_assignment/BUILD index 54b0e3fb87cd1c..455eb131ad5232 100644 --- a/xla/service/memory_space_assignment/BUILD +++ b/xla/service/memory_space_assignment/BUILD @@ -171,7 +171,7 @@ cc_library( "//xla:shape_util", "//xla/hlo/analysis:hlo_alias_analysis", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:instruction_hoister", + "//xla/hlo/transforms/simplifiers:instruction_hoister", "//xla/hlo/utils:hlo_live_range", "//xla/service:buffer_value", "//xla/service:hlo_buffer", diff --git a/xla/service/spmd/BUILD b/xla/service/spmd/BUILD index 8d694310a7acfc..d97584970dee61 100644 --- a/xla/service/spmd/BUILD +++ b/xla/service/spmd/BUILD @@ -55,9 +55,9 @@ cc_library( "//xla/hlo/parser:hlo_lexer", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:flatten_call_graph", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:flatten_call_graph", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/utils:hlo_query", "//xla/hlo/utils:hlo_sharding_util", "//xla/service:call_graph", diff --git a/xla/service/spmd/shardy/BUILD b/xla/service/spmd/shardy/BUILD index 8017dedf3bdbd0..caceb59e3057a5 100644 --- a/xla/service/spmd/shardy/BUILD +++ b/xla/service/spmd/shardy/BUILD @@ -35,8 +35,8 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", - "//xla/hlo/transforms:tuple_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", + "//xla/hlo/transforms/simplifiers:tuple_simplifier", "//xla/hlo/translate/hlo_to_mhlo:hlo_to_mlir_hlo", "//xla/hlo/translate/mhlo_to_hlo:mlir_hlo_to_hlo", "//xla/hlo/utils:hlo_sharding_util", diff --git a/xla/tests/BUILD b/xla/tests/BUILD index 314ebe44bbffc8..31b12e4900b997 100644 --- a/xla/tests/BUILD +++ b/xla/tests/BUILD @@ -494,7 +494,7 @@ xla_test( "//xla:test", "//xla/hlo/builder:xla_computation", "//xla/hlo/transforms:despecializer", - "//xla/hlo/transforms:float_normalization", + "//xla/hlo/transforms/simplifiers:float_normalization", ], ) @@ -514,7 +514,7 @@ xla_test( "//xla:test", "//xla/hlo/builder:xla_computation", "//xla/hlo/transforms:despecializer", - "//xla/hlo/transforms:float_normalization", + "//xla/hlo/transforms/simplifiers:float_normalization", ], ) @@ -539,7 +539,7 @@ xla_test( "//xla:test", "//xla/hlo/builder:xla_computation", "//xla/hlo/transforms:despecializer", - "//xla/hlo/transforms:float_normalization", + "//xla/hlo/transforms/simplifiers:float_normalization", "@com_google_absl//absl/algorithm:container", ], ) @@ -889,7 +889,7 @@ cc_library( "//xla:test", "//xla/hlo/builder:xla_computation", "//xla/hlo/transforms:despecializer", - "//xla/hlo/transforms:float_normalization", + "//xla/hlo/transforms/simplifiers:float_normalization", ], ) @@ -2554,7 +2554,7 @@ xla_test( "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:collective_pipeliner", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", diff --git a/xla/tools/BUILD b/xla/tools/BUILD index a181bd4cff2d80..7ebb2f41306779 100644 --- a/xla/tools/BUILD +++ b/xla/tools/BUILD @@ -163,8 +163,8 @@ cc_library( "//xla:literal_util", "//xla:shape_util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:call_inliner", "//xla/service:compilation_environments", "//xla/service:hlo_module_config", @@ -574,7 +574,7 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:call_graph", "//xla/service:collective_ops_utils", "//xla/service:tuple_util", diff --git a/xla/tools/hlo_bisect/BUILD b/xla/tools/hlo_bisect/BUILD index 8fa673cfd480b2..b4d27ed0065b7d 100644 --- a/xla/tools/hlo_bisect/BUILD +++ b/xla/tools/hlo_bisect/BUILD @@ -46,7 +46,7 @@ cc_library( "//xla:literal_util", "//xla:util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:hlo_dce", + "//xla/hlo/transforms/simplifiers:hlo_dce", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", diff --git a/xla/tools/hlo_opt/BUILD b/xla/tools/hlo_opt/BUILD index 5738942ba889ed..43f2218f573e80 100644 --- a/xla/tools/hlo_opt/BUILD +++ b/xla/tools/hlo_opt/BUILD @@ -84,8 +84,8 @@ cc_library( "//xla:types", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:reduce_window_rewriter", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:reduce_window_rewriter", "//xla/service:buffer_value", "//xla/service:compiler", "//xla/service:dump", @@ -157,9 +157,9 @@ cc_library( "//xla/backends/cpu/codegen:target_machine_features", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:algebraic_simplifier", - "//xla/hlo/transforms:reduce_window_rewriter", "//xla/hlo/transforms:rng_bit_generator_expander", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/transforms/simplifiers:reduce_window_rewriter", "//xla/hlo/translate/hlo_to_mhlo:hlo_to_mlir_hlo", "//xla/service:batchnorm_expander", "//xla/service:change_op_data_type",