From 9faaa8ef23b025c5ef8daa4e40cfbd68a07d4373 Mon Sep 17 00:00:00 2001 From: Shraiysh Date: Mon, 13 Jan 2025 11:57:02 -0800 Subject: [PATCH] PR #21223: [nfc] Cleanup build files for expander transforms Imported from GitHub PR https://github.com/openxla/xla/pull/21223 This is part-3 of #18785 and #20595. 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: -- 7832f8483091b70214d17789c05c38959d67a515 by Shraiysh Vaishay : [nfc] Cleanup build files for expander transforms This is part-3 of #18785 and #20595. Motivation: Smaller build files, fewer merge conflicts, and convinience in development (more intuitive targets). For discussion surrounding this, check thread in #18785. Merging this change closes #21223 FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/21223 from shraiysh:cleanup_expander_transforms 7832f8483091b70214d17789c05c38959d67a515 PiperOrigin-RevId: 715055989 --- xla/backends/interpreter/BUILD | 8 +- xla/hlo/tools/hlo_opt/BUILD | 32 +- xla/hlo/transforms/BUILD | 576 +------------------------- xla/hlo/transforms/collectives/BUILD | 2 +- xla/hlo/transforms/expanders/BUILD | 594 +++++++++++++++++++++++++++ xla/hlo/transforms/simplifiers/BUILD | 6 +- xla/pjrt/interpreter/BUILD | 8 +- xla/service/BUILD | 91 ++-- xla/service/cpu/BUILD | 28 +- xla/service/gpu/BUILD | 34 +- xla/service/gpu/transforms/BUILD | 8 +- xla/service/spmd/BUILD | 2 +- xla/tests/BUILD | 4 +- xla/tools/BUILD | 6 +- xla/tools/hlo_opt/BUILD | 4 +- 15 files changed, 712 insertions(+), 691 deletions(-) create mode 100644 xla/hlo/transforms/expanders/BUILD diff --git a/xla/backends/interpreter/BUILD b/xla/backends/interpreter/BUILD index 85af2b52aac00b..c79b5fca87d12b 100644 --- a/xla/backends/interpreter/BUILD +++ b/xla/backends/interpreter/BUILD @@ -39,10 +39,10 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_module_group", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:cholesky_expander", - "//xla/hlo/transforms:dynamic_index_splitter", - "//xla/hlo/transforms:eigh_expander", - "//xla/hlo/transforms:qr_expander", + "//xla/hlo/transforms/expanders:cholesky_expander", + "//xla/hlo/transforms/expanders:dynamic_index_splitter", + "//xla/hlo/transforms/expanders:eigh_expander", + "//xla/hlo/transforms/expanders:qr_expander", "//xla/service:batchnorm_expander", "//xla/service:compiler", "//xla/service:computation_placer", diff --git a/xla/hlo/tools/hlo_opt/BUILD b/xla/hlo/tools/hlo_opt/BUILD index 490019fa0e96a4..e2a2abe738973d 100644 --- a/xla/hlo/tools/hlo_opt/BUILD +++ b/xla/hlo/tools/hlo_opt/BUILD @@ -34,28 +34,28 @@ cc_library( "//xla/hlo/analysis:indexed_array_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:cholesky_expander", - "//xla/hlo/transforms:comparison_expander", "//xla/hlo/transforms:convert_memory_placement_to_internal_annotations", - "//xla/hlo/transforms:convolution_4d_expander", - "//xla/hlo/transforms:convolution_pred_expander", - "//xla/hlo/transforms:dot_decomposer", - "//xla/hlo/transforms:dynamic_index_splitter", - "//xla/hlo/transforms:eigh_expander", - "//xla/hlo/transforms:logistic_expander", "//xla/hlo/transforms:operand_upcaster", - "//xla/hlo/transforms:optimization_barrier_expander", - "//xla/hlo/transforms:qr_expander", - "//xla/hlo/transforms:real_imag_expander", - "//xla/hlo/transforms:reduce_decomposer", - "//xla/hlo/transforms:reshape_decomposer", - "//xla/hlo/transforms:rng_expander", - "//xla/hlo/transforms:stable_sort_expander", - "//xla/hlo/transforms:stochastic_convert_decomposer", "//xla/hlo/transforms:while_loop_trip_count_annotator", "//xla/hlo/transforms/collectives:all_gather_broadcast_reorder", "//xla/hlo/transforms/collectives:all_reduce_contiguous", "//xla/hlo/transforms/collectives:collective_quantizer", + "//xla/hlo/transforms/expanders:cholesky_expander", + "//xla/hlo/transforms/expanders:comparison_expander", + "//xla/hlo/transforms/expanders:convolution_4d_expander", + "//xla/hlo/transforms/expanders:convolution_pred_expander", + "//xla/hlo/transforms/expanders:dot_decomposer", + "//xla/hlo/transforms/expanders:dynamic_index_splitter", + "//xla/hlo/transforms/expanders:eigh_expander", + "//xla/hlo/transforms/expanders:logistic_expander", + "//xla/hlo/transforms/expanders:optimization_barrier_expander", + "//xla/hlo/transforms/expanders:qr_expander", + "//xla/hlo/transforms/expanders:real_imag_expander", + "//xla/hlo/transforms/expanders:reduce_decomposer", + "//xla/hlo/transforms/expanders:reshape_decomposer", + "//xla/hlo/transforms/expanders:rng_expander", + "//xla/hlo/transforms/expanders:stable_sort_expander", + "//xla/hlo/transforms/expanders:stochastic_convert_decomposer", "//xla/hlo/transforms/simplifiers:all_reduce_folder", "//xla/hlo/transforms/simplifiers:batch_dot_simplification", "//xla/hlo/transforms/simplifiers:broadcast_canonicalizer", diff --git a/xla/hlo/transforms/BUILD b/xla/hlo/transforms/BUILD index c5d95cf3a6f1ae..75b5821f00c16e 100644 --- a/xla/hlo/transforms/BUILD +++ b/xla/hlo/transforms/BUILD @@ -71,329 +71,6 @@ xla_cc_test( ], ) -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:types", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/testlib:pattern_matcher_gmock", - "//xla/hlo/testlib:test", - "//xla/hlo/utils:hlo_matchers", - "//xla/service:hlo_creation_utils", - "//xla/service:pattern_matcher", - "//xla/tsl/lib/core:status_test_util", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_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/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/testlib:test", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - ], -) - -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/hlo/testlib:pattern_matcher_gmock", - "//xla/service:pattern_matcher", - "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:statusor", - ], -) - -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:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/testlib:pattern_matcher_gmock", - "//xla/hlo/testlib:test", - "//xla/service:dynamic_padder", - "//xla/service:pattern_matcher", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", # fixdeps: keep - "@tsl//tsl/platform:statusor", - ], -) - -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", - ], -) - cc_library( name = "while_loop_trip_count_annotator", srcs = ["while_loop_trip_count_annotator.cc"], @@ -478,109 +155,6 @@ xla_cc_test( ], ) -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/testlib:pattern_matcher_gmock", - "//xla/hlo/utils:hlo_matchers", - "//xla/service:pattern_matcher", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", # fixdeps: keep - "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test", - ], -) - -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/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", - ], -) - -xla_cc_test( - name = "reduce_decomposer_test", - srcs = ["expanders/reduce_decomposer_test.cc"], - deps = [ - ":reduce_decomposer", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/testlib:test", - "//xla/hlo/testlib:test_helpers", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", - ], -) - -xla_cc_test( - name = "reshape_decomposer_test", - srcs = ["expanders/reshape_decomposer_test.cc"], - deps = [ - ":reshape_decomposer", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/testlib:test", - "//xla/hlo/testlib:test_helpers", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/strings:string_view", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", - ], -) - cc_library( name = "memory_space_propagation", srcs = ["memory_space_propagation.cc"], @@ -612,41 +186,6 @@ xla_cc_test( ], ) -cc_library( - name = "stable_sort_expander", - srcs = ["expanders/stable_sort_expander.cc"], - hdrs = ["expanders/stable_sort_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "@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:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:string_view", - ], -) - -xla_cc_test( - name = "stable_sort_expander_test", - srcs = ["expanders/stable_sort_expander_test.cc"], - deps = [ - ":stable_sort_expander", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/testlib:pattern_matcher_gmock", - "//xla/hlo/testlib:test", - "//xla/hlo/transforms/simplifiers:algebraic_simplifier", - "//xla/hlo/utils:hlo_matchers", - "//xla/service:pattern_matcher", - "//xla/tsl/lib/core:status_test_util", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", - ], -) - cc_library( name = "convert_memory_placement_to_internal_annotations", srcs = ["convert_memory_placement_to_internal_annotations.cc"], @@ -899,91 +438,16 @@ xla_cc_test( ], ) -cc_library( - name = "dynamic_index_splitter", - srcs = ["expanders/dynamic_index_splitter.cc"], - hdrs = ["expanders/dynamic_index_splitter.h"], - deps = [ - "//xla:shape_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/container:inlined_vector", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:span", - ], -) - -xla_cc_test( - name = "dynamic_index_splitter_test", - srcs = ["expanders/dynamic_index_splitter_test.cc"], - deps = [ - ":dynamic_index_splitter", - "//xla:xla_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/testlib:test", - "//xla/hlo/testlib:test_helpers", - "//xla/hlo/utils:hlo_matchers", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", - ], -) - -cc_library( - name = "rng_expander", - srcs = ["expanders/rng_expander.cc"], - hdrs = ["expanders/rng_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:literal_util", - "//xla:shape_util", - "//xla:xla_data_proto_cc", - "//xla/hlo/builder:xla_builder", - "//xla/hlo/builder/lib:prng", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/container:flat_hash_map", - "@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/synchronization", - ], -) - -cc_library( - name = "rng_bit_generator_expander", - srcs = ["expanders/rng_bit_generator_expander.cc"], - hdrs = ["expanders/rng_bit_generator_expander.h"], - deps = [ - ":op_expander_pass", - "//xla:shape_util", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/builder:xla_builder", - "//xla/hlo/builder/lib:prng", - "//xla/hlo/ir:hlo", - "//xla/service:hlo_creation_utils", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/log:check", - "@com_google_absl//absl/status:statusor", - "@com_google_absl//absl/strings:string_view", - ], -) - cc_library( name = "operand_upcaster", srcs = ["operand_upcaster.cc"], hdrs = ["operand_upcaster.h"], deps = [ - ":op_expander_pass", "//xla:shape_util", "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "//xla/hlo/transforms/expanders:op_expander_pass", "//xla/service:hlo_creation_utils", "//xla/service:shape_inference", "@com_google_absl//absl/algorithm:container", @@ -1011,44 +475,6 @@ xla_cc_test( ], ) -cc_library( - name = "stochastic_convert_decomposer", - srcs = ["expanders/stochastic_convert_decomposer.cc"], - hdrs = ["expanders/stochastic_convert_decomposer.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", - "//xla/service:shape_inference", - "@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", - "@tsl//tsl/platform:statusor", - ], -) - -xla_cc_test( - name = "stochastic_convert_decomposer_test", - srcs = ["expanders/stochastic_convert_decomposer_test.cc"], - deps = [ - ":stochastic_convert_decomposer", - "//xla/hlo/ir:hlo", - "//xla/hlo/parser:hlo_parser", - "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "//xla/hlo/utils:hlo_matchers", - "@com_google_absl//absl/status", - "@com_google_googletest//:gtest", - "@com_google_googletest//:gtest_main", - ], -) - cc_library( name = "sharding_format_picker", testonly = True, diff --git a/xla/hlo/transforms/collectives/BUILD b/xla/hlo/transforms/collectives/BUILD index cf6acd0d3a6aa5..827318f482c5b8 100644 --- a/xla/hlo/transforms/collectives/BUILD +++ b/xla/hlo/transforms/collectives/BUILD @@ -435,7 +435,7 @@ cc_library( deps = [ "//xla:shape_util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "//xla/service:hlo_creation_utils", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", diff --git a/xla/hlo/transforms/expanders/BUILD b/xla/hlo/transforms/expanders/BUILD new file mode 100644 index 00000000000000..b8530c20df5788 --- /dev/null +++ b/xla/hlo/transforms/expanders/BUILD @@ -0,0 +1,594 @@ +# Description: +# Implementation of XLA’s HLO expander transformations. + +load("//xla:xla.bzl", "xla_cc_test") +load("//xla/tsl/platform:rules_cc.bzl", "cc_library") + +package( + # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], + default_visibility = [":friends"], + licenses = ["notice"], +) + +package_group( + name = "friends", + includes = [ + "//xla:friends", + ], +) + +cc_library( + name = "op_expander_pass", + srcs = ["op_expander_pass.cc"], + hdrs = ["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 = ["optimization_barrier_expander.cc"], + hdrs = ["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 = ["comparison_expander.cc"], + hdrs = ["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 = ["cholesky_expander.cc"], + hdrs = ["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 = ["qr_expander.cc"], + hdrs = ["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 = ["real_imag_expander.cc"], + hdrs = ["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 = ["real_imag_expander_test.cc"], + deps = [ + ":real_imag_expander", + "//xla:literal", + "//xla:shape_util", + "//xla:types", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:pattern_matcher_gmock", + "//xla/hlo/testlib:test", + "//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", + "@com_google_googletest//:gtest_main", + ], +) + +cc_library( + name = "eigh_expander", + srcs = ["eigh_expander.cc"], + hdrs = ["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 = ["convolution_4d_expander.cc"], + hdrs = ["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 = ["convolution_4d_expander_test.cc"], + deps = [ + "convolution_4d_expander", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:test", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + ], +) + +cc_library( + name = "convolution_pred_expander", + srcs = ["convolution_pred_expander.cc"], + hdrs = ["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 = ["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/hlo/testlib:pattern_matcher_gmock", + "//xla/service:pattern_matcher", + "@com_google_googletest//:gtest_main", + "@tsl//tsl/platform:statusor", + ], +) + +cc_library( + name = "logistic_expander", + srcs = ["logistic_expander.cc"], + hdrs = ["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 = ["logistic_expander_test.cc"], + deps = [ + ":logistic_expander", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:pattern_matcher_gmock", + "//xla/hlo/testlib:test", + "//xla/service:dynamic_padder", + "//xla/service:pattern_matcher", + "//xla/service:pattern_matcher_gmock", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep + "@tsl//tsl/platform:statusor", + ], +) + +cc_library( + name = "bitcast_dtypes_expander", + srcs = ["bitcast_dtypes_expander.cc"], + hdrs = ["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 = ["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", + ], +) + +cc_library( + name = "dot_decomposer", + srcs = ["dot_decomposer.cc"], + hdrs = ["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 = ["dot_decomposer_test.cc"], + deps = [ + ":dot_decomposer", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:pattern_matcher_gmock", + "//xla/hlo/utils:hlo_matchers", + "//xla/service:pattern_matcher", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep + "@tsl//tsl/platform:statusor", + "@tsl//tsl/platform:test", + ], +) + +cc_library( + name = "reshape_decomposer", + srcs = ["reshape_decomposer.cc"], + hdrs = ["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 = ["reduce_decomposer.cc"], + hdrs = ["reduce_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/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + ], +) + +xla_cc_test( + name = "reduce_decomposer_test", + srcs = ["reduce_decomposer_test.cc"], + deps = [ + ":reduce_decomposer", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:test", + "//xla/hlo/testlib:test_helpers", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", + ], +) + +xla_cc_test( + name = "reshape_decomposer_test", + srcs = ["reshape_decomposer_test.cc"], + deps = [ + ":reshape_decomposer", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:test", + "//xla/hlo/testlib:test_helpers", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", + ], +) + +cc_library( + name = "stable_sort_expander", + srcs = ["stable_sort_expander.cc"], + hdrs = ["stable_sort_expander.h"], + deps = [ + ":op_expander_pass", + "//xla:xla_data_proto_cc", + "//xla/hlo/ir:hlo", + "@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:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:string_view", + ], +) + +xla_cc_test( + name = "stable_sort_expander_test", + srcs = ["stable_sort_expander_test.cc"], + deps = [ + ":stable_sort_expander", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:pattern_matcher_gmock", + "//xla/hlo/testlib:test", + "//xla/hlo/transforms/simplifiers:algebraic_simplifier", + "//xla/hlo/utils:hlo_matchers", + "//xla/service:pattern_matcher", + "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", + ], +) + +cc_library( + name = "dynamic_index_splitter", + srcs = ["dynamic_index_splitter.cc"], + hdrs = ["dynamic_index_splitter.h"], + deps = [ + "//xla:shape_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/container:inlined_vector", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", + ], +) + +xla_cc_test( + name = "dynamic_index_splitter_test", + srcs = ["dynamic_index_splitter_test.cc"], + deps = [ + ":dynamic_index_splitter", + "//xla:xla_proto_cc", + "//xla/hlo/ir:hlo", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/testlib:test", + "//xla/hlo/testlib:test_helpers", + "//xla/hlo/utils:hlo_matchers", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", + ], +) + +cc_library( + name = "rng_expander", + srcs = ["rng_expander.cc"], + hdrs = ["rng_expander.h"], + deps = [ + ":op_expander_pass", + "//xla:literal_util", + "//xla:shape_util", + "//xla:xla_data_proto_cc", + "//xla/hlo/builder:xla_builder", + "//xla/hlo/builder/lib:prng", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@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/synchronization", + ], +) + +cc_library( + name = "rng_bit_generator_expander", + srcs = ["rng_bit_generator_expander.cc"], + hdrs = ["rng_bit_generator_expander.h"], + deps = [ + ":op_expander_pass", + "//xla:shape_util", + "//xla:util", + "//xla:xla_data_proto_cc", + "//xla/hlo/builder:xla_builder", + "//xla/hlo/builder/lib:prng", + "//xla/hlo/ir:hlo", + "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + ], +) + +cc_library( + name = "stochastic_convert_decomposer", + srcs = ["stochastic_convert_decomposer.cc"], + hdrs = ["stochastic_convert_decomposer.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", + "//xla/service:shape_inference", + "@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", + "@tsl//tsl/platform:statusor", + ], +) + +xla_cc_test( + name = "stochastic_convert_decomposer_test", + srcs = ["stochastic_convert_decomposer_test.cc"], + deps = [ + ":stochastic_convert_decomposer", + "//xla/hlo/ir:hlo", + "//xla/hlo/parser:hlo_parser", + "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/status", + "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", + ], +) diff --git a/xla/hlo/transforms/simplifiers/BUILD b/xla/hlo/transforms/simplifiers/BUILD index 23c6f447f447ba..12104b465e753e 100644 --- a/xla/hlo/transforms/simplifiers/BUILD +++ b/xla/hlo/transforms/simplifiers/BUILD @@ -1343,7 +1343,7 @@ cc_library( "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "//xla/service:shape_inference", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -1375,7 +1375,7 @@ cc_library( "//xla:shape_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:errors", @@ -1444,7 +1444,7 @@ cc_library( "//xla:permutation_util", "//xla:shape_util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "//xla/service:gather_scatter_utils", "//xla/service:hlo_creation_utils", "@com_google_absl//absl/algorithm:container", diff --git a/xla/pjrt/interpreter/BUILD b/xla/pjrt/interpreter/BUILD index cbaf62d3624450..5f9c1d597303a7 100644 --- a/xla/pjrt/interpreter/BUILD +++ b/xla/pjrt/interpreter/BUILD @@ -22,10 +22,10 @@ cc_library( "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:cholesky_expander", - "//xla/hlo/transforms:dynamic_index_splitter", - "//xla/hlo/transforms:eigh_expander", - "//xla/hlo/transforms:qr_expander", + "//xla/hlo/transforms/expanders:cholesky_expander", + "//xla/hlo/transforms/expanders:dynamic_index_splitter", + "//xla/hlo/transforms/expanders:eigh_expander", + "//xla/hlo/transforms/expanders:qr_expander", "//xla/pjrt:layout_mode", "//xla/pjrt:mlir_to_hlo", "//xla/pjrt:pjrt_client", diff --git a/xla/service/BUILD b/xla/service/BUILD index e3ccfd8293746a..43d0df748afe6f 100644 --- a/xla/service/BUILD +++ b/xla/service/BUILD @@ -2088,8 +2088,8 @@ cc_library( cc_library( name = "op_expander_pass", hdrs = ["op_expander_pass.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:op_expander_pass instead.", - deps = ["//xla/hlo/transforms:op_expander_pass"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:op_expander_pass instead.", + deps = ["//xla/hlo/transforms/expanders:op_expander_pass"], ) cc_library( @@ -2103,7 +2103,7 @@ cc_library( "//xla:literal_util", "//xla:util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status:statusor", @@ -2114,15 +2114,15 @@ cc_library( cc_library( name = "optimization_barrier_expander", hdrs = ["optimization_barrier_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:optimization_barrier_expander instead.", - deps = ["//xla/hlo/transforms:optimization_barrier_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:optimization_barrier_expander instead.", + deps = ["//xla/hlo/transforms/expanders:optimization_barrier_expander"], ) cc_library( name = "comparison_expander", hdrs = ["comparison_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:comparison_expander instead.", - deps = ["//xla/hlo/transforms:comparison_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:comparison_expander instead.", + deps = ["//xla/hlo/transforms/expanders:comparison_expander"], ) cc_library( @@ -2156,7 +2156,7 @@ cc_library( "//xla:literal_util", "//xla:shape_util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:span", @@ -2178,7 +2178,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/types:span", @@ -2239,7 +2239,7 @@ cc_library( "//xla/hlo/builder/lib:matrix", "//xla/hlo/builder/lib:slicing", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:str_format", @@ -2272,42 +2272,43 @@ xla_cc_test( cc_library( name = "cholesky_expander", hdrs = ["cholesky_expander.h"], - deps = ["//xla/hlo/transforms:cholesky_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:cholesky_expander instead.", + deps = ["//xla/hlo/transforms/expanders:cholesky_expander"], ) cc_library( name = "qr_expander", hdrs = ["qr_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:qr_expander instead.", - deps = ["//xla/hlo/transforms:qr_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:qr_expander instead.", + deps = ["//xla/hlo/transforms/expanders:qr_expander"], ) cc_library( name = "real_imag_expander", hdrs = ["real_imag_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:real_imag_expander instead.", - deps = ["//xla/hlo/transforms:real_imag_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:real_imag_expander instead.", + deps = ["//xla/hlo/transforms/expanders:real_imag_expander"], ) cc_library( name = "eigh_expander", hdrs = ["eigh_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:eigh_expander instead.", - deps = ["//xla/hlo/transforms:eigh_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:eigh_expander instead.", + deps = ["//xla/hlo/transforms/expanders:eigh_expander"], ) cc_library( name = "convolution_4d_expander", hdrs = ["convolution_4d_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:convolution_4d_expander instead.", - deps = ["//xla/hlo/transforms:convolution_4d_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:convolution_4d_expander instead.", + deps = ["//xla/hlo/transforms/expanders:convolution_4d_expander"], ) cc_library( name = "convolution_pred_expander", hdrs = ["convolution_pred_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:convolution_pred_expander instead.", - deps = ["//xla/hlo/transforms:convolution_pred_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:convolution_pred_expander instead.", + deps = ["//xla/hlo/transforms/expanders:convolution_pred_expander"], ) xla_test( @@ -2371,8 +2372,8 @@ cc_library( cc_library( name = "logistic_expander", hdrs = ["logistic_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:logistic_expander instead.", - deps = ["//xla/hlo/transforms:logistic_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:logistic_expander instead.", + deps = ["//xla/hlo/transforms/expanders:logistic_expander"], ) cc_library( @@ -2426,8 +2427,8 @@ cc_library( cc_library( name = "bitcast_dtypes_expander", hdrs = ["bitcast_dtypes_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:bitcast_dtypes_expander instead.", - deps = ["//xla/hlo/transforms:bitcast_dtypes_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:bitcast_dtypes_expander instead.", + deps = ["//xla/hlo/transforms/expanders:bitcast_dtypes_expander"], ) cc_library( @@ -3002,8 +3003,8 @@ cc_library( cc_library( name = "dot_decomposer", hdrs = ["dot_decomposer.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:dot_decomposer instead.", - deps = ["//xla/hlo/transforms:dot_decomposer"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:dot_decomposer instead.", + deps = ["//xla/hlo/transforms/expanders:dot_decomposer"], ) cc_library( @@ -3035,7 +3036,7 @@ cc_library( "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", ], @@ -3096,15 +3097,15 @@ cc_library( cc_library( name = "reshape_decomposer", hdrs = ["reshape_decomposer.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:reshape_decomposer instead.", - deps = ["//xla/hlo/transforms:reshape_decomposer"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:reshape_decomposer instead.", + deps = ["//xla/hlo/transforms/expanders:reshape_decomposer"], ) cc_library( name = "reduce_decomposer", hdrs = ["reduce_decomposer.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:reduce_decomposer instead.", - deps = ["//xla/hlo/transforms:reduce_decomposer"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:reduce_decomposer instead.", + deps = ["//xla/hlo/transforms/expanders:reduce_decomposer"], ) cc_library( @@ -4719,8 +4720,8 @@ cc_library( cc_library( name = "stable_sort_expander", hdrs = ["stable_sort_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:stable_sort_expander instead.", - deps = ["//xla/hlo/transforms:stable_sort_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:stable_sort_expander instead.", + deps = ["//xla/hlo/transforms/expanders:stable_sort_expander"], ) cc_library( @@ -5267,8 +5268,8 @@ cc_library( cc_library( name = "dynamic_index_splitter", hdrs = ["dynamic_index_splitter.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:dynamic_index_splitter instead.", - deps = ["//xla/hlo/transforms:dynamic_index_splitter"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:dynamic_index_splitter instead.", + deps = ["//xla/hlo/transforms/expanders:dynamic_index_splitter"], ) xla_cc_test( @@ -5439,15 +5440,15 @@ cc_library( cc_library( name = "rng_expander", hdrs = ["rng_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:rng_expander instead.", - deps = ["//xla/hlo/transforms:rng_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:rng_expander instead.", + deps = ["//xla/hlo/transforms/expanders:rng_expander"], ) cc_library( name = "rng_bit_generator_expander", hdrs = ["rng_bit_generator_expander.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:rng_bit_generator_expander instead.", - deps = ["//xla/hlo/transforms:rng_bit_generator_expander"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:rng_bit_generator_expander instead.", + deps = ["//xla/hlo/transforms/expanders:rng_bit_generator_expander"], ) cc_library( @@ -5875,7 +5876,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/types:span", @@ -5905,7 +5906,7 @@ cc_library( ":call_inliner", "//xla:literal_util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", ], ) @@ -5985,7 +5986,7 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", @@ -6006,8 +6007,8 @@ cc_library( cc_library( name = "stochastic_convert_decomposer", hdrs = ["stochastic_convert_decomposer.h"], - deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms:stochastic_convert_decomposer instead.", - deps = ["//xla/hlo/transforms:stochastic_convert_decomposer"], + deprecation = "This library is deprecated. Use //third_party/tensorflow/compiler/xla/hlo/transforms/expanders:stochastic_convert_decomposer instead.", + deps = ["//xla/hlo/transforms/expanders:stochastic_convert_decomposer"], ) cc_library( diff --git a/xla/service/cpu/BUILD b/xla/service/cpu/BUILD index 293901687491d2..a01b3dcf670266 100644 --- a/xla/service/cpu/BUILD +++ b/xla/service/cpu/BUILD @@ -255,23 +255,23 @@ cc_library( "//xla/hlo/ir:hlo_module_group", "//xla/hlo/pass:hlo_pass", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:bitcast_dtypes_expander", - "//xla/hlo/transforms:cholesky_expander", - "//xla/hlo/transforms:comparison_expander", - "//xla/hlo/transforms:dot_decomposer", - "//xla/hlo/transforms:dynamic_index_splitter", - "//xla/hlo/transforms:eigh_expander", "//xla/hlo/transforms:literal_canonicalizer", - "//xla/hlo/transforms:logistic_expander", "//xla/hlo/transforms:operand_upcaster", - "//xla/hlo/transforms:optimization_barrier_expander", - "//xla/hlo/transforms:qr_expander", - "//xla/hlo/transforms:reduce_decomposer", - "//xla/hlo/transforms:reshape_decomposer", - "//xla/hlo/transforms:rng_bit_generator_expander", - "//xla/hlo/transforms:rng_expander", - "//xla/hlo/transforms:stochastic_convert_decomposer", "//xla/hlo/transforms:while_loop_trip_count_annotator", + "//xla/hlo/transforms/expanders:bitcast_dtypes_expander", + "//xla/hlo/transforms/expanders:cholesky_expander", + "//xla/hlo/transforms/expanders:comparison_expander", + "//xla/hlo/transforms/expanders:dot_decomposer", + "//xla/hlo/transforms/expanders:dynamic_index_splitter", + "//xla/hlo/transforms/expanders:eigh_expander", + "//xla/hlo/transforms/expanders:logistic_expander", + "//xla/hlo/transforms/expanders:optimization_barrier_expander", + "//xla/hlo/transforms/expanders:qr_expander", + "//xla/hlo/transforms/expanders:reduce_decomposer", + "//xla/hlo/transforms/expanders:reshape_decomposer", + "//xla/hlo/transforms/expanders:rng_bit_generator_expander", + "//xla/hlo/transforms/expanders:rng_expander", + "//xla/hlo/transforms/expanders:stochastic_convert_decomposer", "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/hlo/transforms/simplifiers:batch_dot_simplification", "//xla/hlo/transforms/simplifiers:broadcast_canonicalizer", diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD index e7bceb8990efc9..f577dafac5dfc1 100644 --- a/xla/service/gpu/BUILD +++ b/xla/service/gpu/BUILD @@ -1443,6 +1443,23 @@ 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/expanders:bitcast_dtypes_expander", + "//xla/hlo/transforms/expanders:comparison_expander", + "//xla/hlo/transforms/expanders:convolution_4d_expander", + "//xla/hlo/transforms/expanders:convolution_pred_expander", + "//xla/hlo/transforms/expanders:dot_decomposer", + "//xla/hlo/transforms/expanders:dynamic_index_splitter", + "//xla/hlo/transforms/expanders:eigh_expander", + "//xla/hlo/transforms/expanders:logistic_expander", + "//xla/hlo/transforms/expanders:optimization_barrier_expander", + "//xla/hlo/transforms/expanders:qr_expander", + "//xla/hlo/transforms/expanders:real_imag_expander", + "//xla/hlo/transforms/expanders:reduce_decomposer", + "//xla/hlo/transforms/expanders:reshape_decomposer", + "//xla/hlo/transforms/expanders:rng_bit_generator_expander", + "//xla/hlo/transforms/expanders:rng_expander", + "//xla/hlo/transforms/expanders:stable_sort_expander", + "//xla/hlo/transforms/expanders:stochastic_convert_decomposer", "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/hlo/transforms/simplifiers:all_reduce_folder", "//xla/hlo/transforms/simplifiers:broadcast_canonicalizer", @@ -1468,27 +1485,10 @@ cc_library( "//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:comparison_expander", "//xla/hlo/transforms:convert_memory_placement_to_internal_annotations", - "//xla/hlo/transforms:convolution_4d_expander", - "//xla/hlo/transforms:convolution_pred_expander", - "//xla/hlo/transforms:dot_decomposer", - "//xla/hlo/transforms:dynamic_index_splitter", - "//xla/hlo/transforms:eigh_expander", "//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:qr_expander", - "//xla/hlo/transforms:real_imag_expander", - "//xla/hlo/transforms:reduce_decomposer", - "//xla/hlo/transforms:reshape_decomposer", - "//xla/hlo/transforms:rng_bit_generator_expander", - "//xla/hlo/transforms:rng_expander", - "//xla/hlo/transforms:stable_sort_expander", - "//xla/hlo/transforms:stochastic_convert_decomposer", "//xla/hlo/transforms:while_loop_trip_count_annotator", "//xla/hlo/translate/hlo_to_mhlo:hlo_utils", "//xla/hlo/translate/mhlo_to_hlo:location_exporter", diff --git a/xla/service/gpu/transforms/BUILD b/xla/service/gpu/transforms/BUILD index 1cf97b4ff0664c..0cf3c1bf7dc3e6 100644 --- a/xla/service/gpu/transforms/BUILD +++ b/xla/service/gpu/transforms/BUILD @@ -370,7 +370,7 @@ cc_library( hdrs = ["all_gather_dynamic_slice_simplifier.h"], deps = [ "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "//xla/service:collective_opt_utils", "//xla/service:hlo_module_config", "@com_google_absl//absl/status:statusor", @@ -997,7 +997,7 @@ xla_test( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", - "//xla/hlo/transforms:reshape_decomposer", + "//xla/hlo/transforms/expanders:reshape_decomposer", "//xla/hlo/transforms/simplifiers:algebraic_simplifier", "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/service:computation_layout", @@ -1416,7 +1416,7 @@ cc_library( "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:errors", @@ -1447,7 +1447,7 @@ cc_library( "//xla:shape_util", "//xla:util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:op_expander_pass", + "//xla/hlo/transforms/expanders:op_expander_pass", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:errors", diff --git a/xla/service/spmd/BUILD b/xla/service/spmd/BUILD index 07781581968e12..407b6280d4d026 100644 --- a/xla/service/spmd/BUILD +++ b/xla/service/spmd/BUILD @@ -274,7 +274,7 @@ xla_cc_test( "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:rng_expander", + "//xla/hlo/transforms/expanders:rng_expander", "//xla/hlo/utils:hlo_matchers", "//xla/service:hlo_module_config", "//xla/service:hlo_verifier", diff --git a/xla/tests/BUILD b/xla/tests/BUILD index 60eb15d5a54645..68e3a158beb494 100644 --- a/xla/tests/BUILD +++ b/xla/tests/BUILD @@ -2259,8 +2259,8 @@ xla_test( "//xla:literal", "//xla:literal_util", "//xla/hlo/ir:hlo", - "//xla/hlo/transforms:rng_bit_generator_expander", - "//xla/hlo/transforms:rng_expander", + "//xla/hlo/transforms/expanders:rng_bit_generator_expander", + "//xla/hlo/transforms/expanders:rng_expander", "@com_google_absl//absl/status", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", diff --git a/xla/tools/BUILD b/xla/tools/BUILD index ca195b1b4ab8a9..498386a0114511 100644 --- a/xla/tools/BUILD +++ b/xla/tools/BUILD @@ -228,9 +228,9 @@ cc_library( deps = [ "//xla:xla_data_proto_cc", "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/hlo/transforms:cholesky_expander", - "//xla/hlo/transforms:rng_bit_generator_expander", - "//xla/hlo/transforms:rng_expander", + "//xla/hlo/transforms/expanders:cholesky_expander", + "//xla/hlo/transforms/expanders:rng_bit_generator_expander", + "//xla/hlo/transforms/expanders:rng_expander", "//xla/service:batchnorm_expander", "//xla/service:hlo_proto_cc", "//xla/service:hlo_verifier", diff --git a/xla/tools/hlo_opt/BUILD b/xla/tools/hlo_opt/BUILD index 43f2218f573e80..ccf3947d681430 100644 --- a/xla/tools/hlo_opt/BUILD +++ b/xla/tools/hlo_opt/BUILD @@ -34,7 +34,7 @@ cc_library( "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/tools/hlo_opt:opt_lib", - "//xla/hlo/transforms:bitcast_dtypes_expander", + "//xla/hlo/transforms/expanders:bitcast_dtypes_expander", "//xla/service:all_reduce_simplifier", "//xla/service:all_to_all_decomposer", "//xla/service:batched_gather_scatter_normalizer", @@ -157,7 +157,7 @@ cc_library( "//xla/backends/cpu/codegen:target_machine_features", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", - "//xla/hlo/transforms:rng_bit_generator_expander", + "//xla/hlo/transforms/expanders: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",