diff --git a/third_party/xla/xla/tools/BUILD b/third_party/xla/xla/tools/BUILD index 06c37dddab3401..a6e7eed3123976 100644 --- a/third_party/xla/xla/tools/BUILD +++ b/third_party/xla/xla/tools/BUILD @@ -13,25 +13,25 @@ load( "if_cuda_is_configured", ) load( - "//xla:xla.bzl", + "@local_xla//xla:xla.bzl", "xla_cc_binary", "xla_cc_test", "xla_internal", "xla_py_proto_library", ) -load("//xla/stream_executor:build_defs.bzl", "if_gpu_is_configured") -load("//xla/tests:build_defs.bzl", "xla_test") +load("@local_xla//xla/stream_executor:build_defs.bzl", "if_gpu_is_configured") +load("@local_xla//xla/tests:build_defs.bzl", "xla_test") load( - "//xla/tsl:tsl.bzl", + "@local_xla//xla/tsl:tsl.bzl", "if_cuda_or_rocm", "if_google", "tsl_gpu_library", ) -load("//xla/tsl:tsl.default.bzl", "filegroup") +load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup") package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], - default_visibility = ["//xla:internal"], + default_visibility = ["@local_xla//xla:internal"], licenses = ["notice"], ) @@ -42,7 +42,7 @@ filegroup( "**/*.cc", "**/*.h", ]), - visibility = ["//xla:internal"], + visibility = ["@local_xla//xla:internal"], ) build_test( @@ -56,9 +56,9 @@ xla_cc_binary( name = "hex_floats_to_packed_literal", srcs = ["hex_floats_to_packed_literal.cc"], deps = [ - "//xla/tsl/lib/io:buffered_inputstream", - "//xla/tsl/lib/io:random_inputstream", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla/tsl/lib/io:buffered_inputstream", + "@local_xla//xla/tsl/lib/io:random_inputstream", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/base", "@com_google_absl//absl/strings", "@local_tsl//tsl/platform:env", @@ -79,13 +79,13 @@ xla_cc_binary( name = "show_signature", srcs = ["show_signature.cc"], deps = [ - "//xla:shape_util", - "//xla:types", - "//xla/client", - "//xla/client:client_library", - "//xla/client:local_client", - "//xla/service:hlo_proto_cc", - "//xla/service:interpreter_plugin", + "@local_xla//xla:shape_util", + "@local_xla//xla:types", + "@local_xla//xla/client", + "@local_xla//xla/client:client_library", + "@local_xla//xla/client:local_client", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/service:interpreter_plugin", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:span", "@local_tsl//tsl/platform:env", @@ -105,9 +105,9 @@ xla_cc_binary( name = "show_literal", srcs = ["show_literal.cc"], deps = [ - "//xla:literal", - "//xla:types", - "//xla:xla_data_proto_cc", + "@local_xla//xla:literal", + "@local_xla//xla:types", + "@local_xla//xla:xla_data_proto_cc", "@local_tsl//tsl/platform:env", "@local_tsl//tsl/platform:logging", "@local_tsl//tsl/platform:platform_port", @@ -126,7 +126,7 @@ xla_cc_binary( name = "convert_computation", srcs = ["convert_computation.cc"], deps = [ - "//xla/service:hlo_proto_cc", + "@local_xla//xla/service:hlo_proto_cc", "@local_tsl//tsl/platform:env", "@local_tsl//tsl/platform:logging", "@local_tsl//tsl/platform:platform_port", @@ -146,7 +146,7 @@ xla_cc_binary( name = "hlo_module_metadata_processor", srcs = ["hlo_module_metadata_processor.cc"], deps = [ - "//xla/service:hlo_proto_cc", + "@local_xla//xla/service:hlo_proto_cc", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", @@ -169,10 +169,10 @@ xla_cc_binary( name = "show_text_literal", srcs = ["show_text_literal.cc"], deps = [ - "//xla:literal", - "//xla:text_literal_reader", - "//xla:types", - "//xla:xla_data_proto_cc", + "@local_xla//xla:literal", + "@local_xla//xla:text_literal_reader", + "@local_xla//xla:types", + "@local_xla//xla:xla_data_proto_cc", "@com_google_absl//absl/status:statusor", "@local_tsl//tsl/platform:logging", "@local_tsl//tsl/platform:platform_port", @@ -191,18 +191,18 @@ xla_cc_binary( name = "dumped_computation_to_text", srcs = ["dumped_computation_to_text.cc"], deps = [ - "//xla:shape_util", - "//xla:xla_proto_cc", - "//xla/client:client_library", - "//xla/client:executable_build_options", - "//xla/client:local_client", - "//xla/client:xla_computation", - "//xla/hlo/ir:hlo", - "//xla/service", - "//xla/service:hlo_proto_cc", - "//xla/service:interpreter_plugin", - "//xla/service:local_service", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla:shape_util", + "@local_xla//xla:xla_proto_cc", + "@local_xla//xla/client:client_library", + "@local_xla//xla/client:executable_build_options", + "@local_xla//xla/client:local_client", + "@local_xla//xla/client:xla_computation", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/service:interpreter_plugin", + "@local_xla//xla/service:local_service", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:span", "@local_tsl//tsl/platform:env", @@ -223,16 +223,16 @@ xla_cc_binary( name = "dumped_computation_to_operation_list", srcs = ["dumped_computation_to_operation_list.cc"], deps = [ - "//xla:shape_util", - "//xla/client:client_library", - "//xla/client:executable_build_options", - "//xla/client:local_client", - "//xla/client:xla_computation", - "//xla/hlo/ir:hlo", - "//xla/service", - "//xla/service:hlo_proto_cc", - "//xla/service:interpreter_plugin", - "//xla/service:local_service", + "@local_xla//xla:shape_util", + "@local_xla//xla/client:client_library", + "@local_xla//xla/client:executable_build_options", + "@local_xla//xla/client:local_client", + "@local_xla//xla/client:xla_computation", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/service:interpreter_plugin", + "@local_xla//xla/service:local_service", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -256,9 +256,9 @@ xla_cc_binary( name = "hlo_proto_to_json", srcs = ["hlo_proto_to_json.cc"], deps = [ - "//xla:util", - "//xla/service:hlo_proto_cc", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla:util", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/status:statusor", "@local_tsl//tsl/platform:env", "@local_tsl//tsl/platform:logging", @@ -272,10 +272,10 @@ xla_cc_test( srcs = ["hlo_extractor_test.cc"], deps = [ ":hlo_extractor", - "//xla/hlo/ir:hlo", - "//xla/hlo/utils:hlo_matchers", - "//xla/tests:hlo_test_base", - "//xla/tests:xla_internal_test_main", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/hlo/utils:hlo_matchers", + "@local_xla//xla/tests:hlo_test_base", + "@local_xla//xla/tests:xla_internal_test_main", "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:statusor", ], @@ -286,14 +286,14 @@ cc_library( srcs = ["hlo_extractor.cc"], hdrs = ["hlo_extractor.h"], deps = [ - "//xla:literal", - "//xla:literal_util", - "//xla:shape_util", - "//xla/hlo/ir:hlo", - "//xla/service:compilation_environments", - "//xla/service:hlo_module_config", - "//xla/service:hlo_verifier", - "//xla/tests:test_utils", + "@local_xla//xla:literal", + "@local_xla//xla:literal_util", + "@local_xla//xla:shape_util", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:compilation_environments", + "@local_xla//xla/service:hlo_module_config", + "@local_xla//xla/service:hlo_verifier", + "@local_xla//xla/tests:test_utils", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log:check", @@ -316,7 +316,7 @@ xla_cc_binary( testonly = True, linkopts = ["-Wl,-rpath,$$ORIGIN/../lit_lib"], deps = [ - "//xla/tools/hlo_opt:opt_main", + "@local_xla//xla/tools/hlo_opt:opt_main", ], ) @@ -326,10 +326,10 @@ cc_library( deps = [ ":hlo_expand_lib", ":hlo_module_loader", - "//xla:xla_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla:xla_proto_cc", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/hlo/pass:hlo_pass_pipeline", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -344,18 +344,18 @@ cc_library( srcs = ["hlo_expand.cc"], hdrs = ["hlo_expand.h"], deps = [ - "//xla:xla_data_proto_cc", - "//xla/hlo/pass:hlo_pass_pipeline", - "//xla/service:batchnorm_expander", - "//xla/service:cholesky_expander", - "//xla/service:hlo_proto_cc", - "//xla/service:hlo_verifier", - "//xla/service:rng_bit_generator_expander", - "//xla/service:rng_expander", - "//xla/service:sharding_propagation", - "//xla/service:triangular_solve_expander", - "//xla/service/spmd:stateful_rng_spmd_partitioner", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla:xla_data_proto_cc", + "@local_xla//xla/hlo/pass:hlo_pass_pipeline", + "@local_xla//xla/service:batchnorm_expander", + "@local_xla//xla/service:cholesky_expander", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/service:hlo_verifier", + "@local_xla//xla/service:rng_bit_generator_expander", + "@local_xla//xla/service:rng_expander", + "@local_xla//xla/service:sharding_propagation", + "@local_xla//xla/service:triangular_solve_expander", + "@local_xla//xla/service/spmd:stateful_rng_spmd_partitioner", + "@local_xla//xla/tsl/util:command_line_flags", ], ) @@ -385,10 +385,10 @@ xla_cc_test( srcs = ["hlo_slicer_test.cc"], deps = [ ":hlo_slicer", - "//xla/hlo/ir:hlo", - "//xla/hlo/utils:hlo_matchers", - "//xla/tests:hlo_test_base", - "//xla/tests:xla_internal_test_main", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/hlo/utils:hlo_matchers", + "@local_xla//xla/tests:hlo_test_base", + "@local_xla//xla/tests:xla_internal_test_main", "@com_google_absl//absl/log:check", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", @@ -402,10 +402,10 @@ cc_library( hdrs = ["hlo_slicer.h"], deps = [ ":hlo_extractor", - "//xla:shape_util", - "//xla/hlo/ir:hlo", - "//xla/service:call_graph", - "//xla/service:hlo_verifier", + "@local_xla//xla:shape_util", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:call_graph", + "@local_xla//xla/service:hlo_verifier", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", @@ -420,15 +420,15 @@ xla_cc_binary( srcs = ["interactive_graphviz.cc"], deps = [ ":hlo_extractor", - "//xla/client:client_library", - "//xla/client:local_client", - "//xla/service:compiler", - "//xla/service:cpu_plugin", - "//xla/service:hlo_proto_cc", - "//xla/service:hlo_runner", - "//xla/service:local_service", - "//xla/service:platform_util", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla/client:client_library", + "@local_xla//xla/client:local_client", + "@local_xla//xla/service:compiler", + "@local_xla//xla/service:cpu_plugin", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/service:hlo_runner", + "@local_xla//xla/service:local_service", + "@local_xla//xla/service:platform_util", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/strings", @@ -438,12 +438,12 @@ xla_cc_binary( "@local_tsl//tsl/platform:subprocess", "@local_tsl//tsl/protobuf:error_codes_proto_impl_cc", ] + if_cuda_or_rocm([ - "//xla/service:gpu_plugin", + "@local_xla//xla/service:gpu_plugin", ]) + if_cuda([ - "//xla/stream_executor:cuda_platform", + "@local_xla//xla/stream_executor:cuda_platform", ]) + xla_internal( ["service:hlo_graph_dumper_google"], - otherwise = ["//xla/service:hlo_graph_dumper"], + otherwise = ["@local_xla//xla/service:hlo_graph_dumper"], ), ) @@ -468,13 +468,13 @@ cc_library( name = "hlo_module_loader", srcs = ["hlo_module_loader.cc"], hdrs = ["hlo_module_loader.h"], - visibility = ["//xla:friends"], + visibility = ["@local_xla//xla:friends"], deps = [ ":run_hlo_module_proto_cc", - "//xla:debug_options_flags", - "//xla/hlo/ir:hlo", - "//xla/service:hlo_module_config", - "//xla/service:hlo_parser", + "@local_xla//xla:debug_options_flags", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:hlo_module_config", + "@local_xla//xla/service:hlo_parser", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@local_tsl//tsl/platform:env", @@ -489,9 +489,9 @@ xla_cc_test( srcs = ["hlo_module_loader_test.cc"], deps = [ ":hlo_module_loader", - "//xla/tests:hlo_test_base", - "//xla/tests:xla_internal_test_main", # fixdeps: keep - "//xla/tsl/lib/core:status_test_util", + "@local_xla//xla/tests:hlo_test_base", + "@local_xla//xla/tests:xla_internal_test_main", # fixdeps: keep + "@local_xla//xla/tsl/lib/core:status_test_util", "@local_tsl//tsl/platform:test", ], ) @@ -501,13 +501,13 @@ cc_library( srcs = ["prepare_reference_module.cc"], hdrs = ["prepare_reference_module.h"], deps = [ - "//xla:debug_options_flags", - "//xla:xla_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/service:despecializer", - "//xla/service:hlo_module_config", - "//xla/service:hlo_runner_interface", - "//xla/stream_executor:platform", + "@local_xla//xla:debug_options_flags", + "@local_xla//xla:xla_proto_cc", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:despecializer", + "@local_xla//xla/service:hlo_module_config", + "@local_xla//xla/service:hlo_runner_interface", + "@local_xla//xla/stream_executor:platform", "@com_google_absl//absl/status:statusor", "@local_tsl//tsl/platform:errors", "@local_tsl//tsl/platform:status", @@ -519,9 +519,9 @@ xla_cc_test( srcs = ["prepare_reference_module_test.cc"], deps = [ ":prepare_reference_module", - "//xla:test", - "//xla/hlo/ir:hlo", - "//xla/tests:hlo_test_base", + "@local_xla//xla:test", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/tests:hlo_test_base", "@local_tsl//tsl/platform:statusor", "@local_tsl//tsl/platform:test_main", ], @@ -531,7 +531,7 @@ tf_proto_library( name = "run_hlo_module_proto", srcs = ["run_hlo_module.proto"], protodeps = [ - "//xla:xla_data_proto", + "@local_xla//xla:xla_data_proto", ], visibility = ["//visibility:public"], ) @@ -548,9 +548,9 @@ cc_library( srcs = ["hlo_decomposer.cc"], hdrs = ["hlo_decomposer.h"], deps = [ - "//xla/hlo/ir:hlo", - "//xla/service:call_graph", - "//xla/service:compilation_environments", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:call_graph", + "@local_xla//xla/service:compilation_environments", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/container:inlined_vector", @@ -571,17 +571,17 @@ cc_library( ":hlo_module_loader", ":prepare_reference_module", ":run_hlo_module_proto_cc", - "//xla:error_spec", - "//xla:literal", - "//xla:literal_comparison", - "//xla:util", - "//xla:xla_data_proto_cc", - "//xla/hlo/ir:hlo", - "//xla/service:hlo_module_config", - "//xla/service:hlo_proto_cc", - "//xla/service:hlo_runner", - "//xla/service:hlo_verifier", - "//xla/tests:test_utils", + "@local_xla//xla:error_spec", + "@local_xla//xla:literal", + "@local_xla//xla:literal_comparison", + "@local_xla//xla:util", + "@local_xla//xla:xla_data_proto_cc", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:hlo_module_config", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/service:hlo_runner", + "@local_xla//xla/service:hlo_verifier", + "@local_xla//xla/tests:test_utils", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", @@ -602,10 +602,10 @@ xla_cc_test( deps = [ ":run_hlo_module_lib", ":run_hlo_module_proto_cc", - "//xla:literal", - "//xla:literal_util", - "//xla:xla_data_proto_cc", - "//xla/tsl/lib/core:status_test_util", + "@local_xla//xla:literal", + "@local_xla//xla:literal_util", + "@local_xla//xla:xla_data_proto_cc", + "@local_xla//xla/tsl/lib/core:status_test_util", "@local_tsl//tsl/platform:env", "@local_tsl//tsl/platform:test", "@local_tsl//tsl/platform:test_main", @@ -621,15 +621,15 @@ xla_cc_binary( ], deps = [ ":run_hlo_module_lib", - "//xla:debug_options_flags", - "//xla/hlo/translate/mhlo_to_hlo:translate", - "//xla/hlo/translate/stablehlo_to_hlo:translate", - "//xla/service:cpu_plugin", - "//xla/service:hlo_module_config", - "//xla/service:hlo_runner", - "//xla/service:interpreter_plugin", - "//xla/service:platform_util", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla:debug_options_flags", + "@local_xla//xla/hlo/translate/mhlo_to_hlo:translate", + "@local_xla//xla/hlo/translate/stablehlo_to_hlo:translate", + "@local_xla//xla/service:cpu_plugin", + "@local_xla//xla/service:hlo_module_config", + "@local_xla//xla/service:hlo_runner", + "@local_xla//xla/service:interpreter_plugin", + "@local_xla//xla/service:platform_util", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/strings", "@llvm-project//llvm:Support", "@local_tsl//tsl/platform:logging", @@ -638,9 +638,9 @@ xla_cc_binary( "@local_tsl//tsl/platform:status", "@local_tsl//tsl/platform:test", ] + if_cuda_or_rocm([ - "//xla/service:gpu_plugin", + "@local_xla//xla/service:gpu_plugin", ]) + if_cuda([ - "//xla/stream_executor:cuda_platform", + "@local_xla//xla/stream_executor:cuda_platform", ]), ) @@ -657,11 +657,11 @@ xla_cc_test( ":run_hlo_module", ], deps = [ - "//xla:literal", - "//xla:literal_util", - "//xla/hlo/ir:hlo", - "//xla/service:hlo_parser", - "//xla/tsl/lib/core:status_test_util", + "@local_xla//xla:literal", + "@local_xla//xla:literal_util", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:hlo_parser", + "@local_xla//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/log", "@com_google_absl//absl/strings", "@local_tsl//tsl/platform:env", @@ -678,17 +678,17 @@ cc_library( srcs = ["hlo_control_flow_flattening.cc"], hdrs = ["hlo_control_flow_flattening.h"], deps = [ - "//xla:comparison_util", - "//xla:literal", - "//xla:literal_util", - "//xla:shape_util", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/hlo/pass:hlo_pass", - "//xla/service:call_graph", - "//xla/service:collective_ops_utils", - "//xla/service:hlo_dce", - "//xla/service:tuple_util", + "@local_xla//xla:comparison_util", + "@local_xla//xla:literal", + "@local_xla//xla:literal_util", + "@local_xla//xla:shape_util", + "@local_xla//xla:util", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/hlo/pass:hlo_pass", + "@local_xla//xla/service:call_graph", + "@local_xla//xla/service:collective_ops_utils", + "@local_xla//xla/service:hlo_dce", + "@local_xla//xla/service:tuple_util", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", @@ -705,14 +705,14 @@ xla_cc_test( srcs = ["hlo_control_flow_flattening_test.cc"], deps = [ ":hlo_control_flow_flattening", - "//xla/hlo/utils:hlo_matchers", - "//xla/service:collective_ops_utils", - "//xla/service:despecializer", - "//xla/service:hlo_verifier", - "//xla/service/spmd:spmd_partitioner", - "//xla/tests:hlo_test_base", - "//xla/tests:xla_internal_test_main", # fixdeps: keep - "//xla/tsl/lib/core:status_test_util", + "@local_xla//xla/hlo/utils:hlo_matchers", + "@local_xla//xla/service:collective_ops_utils", + "@local_xla//xla/service:despecializer", + "@local_xla//xla/service:hlo_verifier", + "@local_xla//xla/service/spmd:spmd_partitioner", + "@local_xla//xla/tests:hlo_test_base", + "@local_xla//xla/tests:xla_internal_test_main", # fixdeps: keep + "@local_xla//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings", ], ) @@ -738,10 +738,10 @@ xla_cc_binary( srcs = ["compute_cost.cc"], deps = [ ":hlo_module_loader", - "//xla:debug_options_flags", - "//xla:shape_util", - "//xla/service:hlo_cost_analysis", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla:debug_options_flags", + "@local_xla//xla:shape_util", + "@local_xla//xla/service:hlo_cost_analysis", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/log", "@com_google_absl//absl/strings", "@local_tsl//tsl/platform:platform_port", @@ -755,10 +755,10 @@ xla_cc_binary( deps = [ ":hlo_decomposer_lib", ":hlo_module_loader", - "//xla:debug_options_flags", - "//xla/hlo/ir:hlo", - "//xla/service:hlo_proto_cc", - "//xla/tsl/util:command_line_flags", + "@local_xla//xla:debug_options_flags", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/tsl/util:command_line_flags", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", @@ -779,26 +779,26 @@ tsl_gpu_library( visibility = ["//visibility:public"], deps = [ ":hlo_module_loader", - "//xla:debug_options_flags", - "//xla:shape_util", - "//xla:util", - "//xla/client:xla_computation", - "//xla/hlo/ir:hlo", - "//xla/hlo/ir:hlo_module_group", - "//xla/mlir_hlo", - "//xla/pjrt:mlir_to_hlo", - "//xla/service:compiler", - "//xla/service:executable", - "//xla/service:export_hlo", - "//xla/service:hlo_module_config", - "//xla/service:hlo_proto_cc", - "//xla/service:symbol_repository", - "//xla/service:xla_compile_result_proto_cc_impl", - "//xla/service/cpu:cpu_compiler", - "//xla/service/cpu:cpu_executable", - "//xla/stream_executor", - "//xla/stream_executor:device_memory_allocator", - "//xla/stream_executor:stream_executor_memory_allocator", + "@local_xla//xla:debug_options_flags", + "@local_xla//xla:shape_util", + "@local_xla//xla:util", + "@local_xla//xla/client:xla_computation", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/hlo/ir:hlo_module_group", + "@local_xla//xla/mlir_hlo", + "@local_xla//xla/pjrt:mlir_to_hlo", + "@local_xla//xla/service:compiler", + "@local_xla//xla/service:executable", + "@local_xla//xla/service:export_hlo", + "@local_xla//xla/service:hlo_module_config", + "@local_xla//xla/service:hlo_proto_cc", + "@local_xla//xla/service:symbol_repository", + "@local_xla//xla/service:xla_compile_result_proto_cc_impl", + "@local_xla//xla/service/cpu:cpu_compiler", + "@local_xla//xla/service/cpu:cpu_executable", + "@local_xla//xla/stream_executor", + "@local_xla//xla/stream_executor:device_memory_allocator", + "@local_xla//xla/stream_executor:stream_executor_memory_allocator", "@com_google_absl//absl/cleanup", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", @@ -818,17 +818,17 @@ tsl_gpu_library( "@local_tsl//tsl/platform:statusor", "@stablehlo//:register", ] + if_cuda_is_configured([ - "//xla/service/gpu:nvptx_compiler", - "//xla/service/gpu:nvptx_compiler_impl", + "@local_xla//xla/service/gpu:nvptx_compiler", + "@local_xla//xla/service/gpu:nvptx_compiler_impl", ]) + if_rocm_is_configured([ - "//xla/service/gpu:amdgpu_compiler", - "//xla/service/gpu:amdgpu_compiler_impl", + "@local_xla//xla/service/gpu:amdgpu_compiler", + "@local_xla//xla/service/gpu:amdgpu_compiler_impl", ]) + if_gpu_is_configured([ - "//xla/service/gpu:executable_proto_cc", - "//xla/service/gpu:gpu_compiler", - "//xla/service/gpu/autotuning:autotuner_util", - "//xla/stream_executor/gpu:gpu_init", - "//xla/service/gpu:gpu_symbol_repository", + "@local_xla//xla/service/gpu:executable_proto_cc", + "@local_xla//xla/service/gpu:gpu_compiler", + "@local_xla//xla/service/gpu/autotuning:autotuner_util", + "@local_xla//xla/stream_executor/gpu:gpu_init", + "@local_xla//xla/service/gpu:gpu_symbol_repository", ]) + if_google(["@com_google_protobuf//:duration_cc_proto"]), ) @@ -843,14 +843,14 @@ xla_test( ], deps = [ ":xla_compile_lib", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/service:platform_util", - "//xla/service:symbol_repository", - "//xla/service:xla_compile_result_proto_cc_impl", - "//xla/tests:hlo_test_base", - "//xla/tests:xla_internal_test_main", # fixdeps: keep - "//xla/tsl/lib/core:status_test_util", + "@local_xla//xla:util", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:platform_util", + "@local_xla//xla/service:symbol_repository", + "@local_xla//xla/service:xla_compile_result_proto_cc_impl", + "@local_xla//xla/tests:hlo_test_base", + "@local_xla//xla/tests:xla_internal_test_main", # fixdeps: keep + "@local_xla//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings", "@com_google_absl//absl/synchronization", "@com_google_googletest//:gtest", @@ -874,8 +874,8 @@ xla_test( ], data = [ ":data/add.hlo", - "//xla/service:xla_aot_compile_test_gpu_target_config.prototxt", - "//xla/service/gpu:gpu_compiler_test_autotune_db.textproto", + "@local_xla//xla/service:xla_aot_compile_test_gpu_target_config.prototxt", + "@local_xla//xla/service/gpu:gpu_compiler_test_autotune_db.textproto", ], tags = [ "config-cuda-only", @@ -883,17 +883,17 @@ xla_test( ], deps = [ ":xla_compile_lib", - "//xla:util", - "//xla/hlo/ir:hlo", - "//xla/service:platform_util", - "//xla/service:symbol_repository", - "//xla/service:xla_compile_result_proto_cc_impl", - "//xla/service/gpu:gpu_symbol_repository", - "//xla/service/gpu/autotuning:autotuner_util", - "//xla/stream_executor:device_description_proto_cc", - "//xla/tests:hlo_test_base", - "//xla/tests:xla_internal_test_main", # fixdeps: keep - "//xla/tsl/lib/core:status_test_util", + "@local_xla//xla:util", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/service:platform_util", + "@local_xla//xla/service:symbol_repository", + "@local_xla//xla/service:xla_compile_result_proto_cc_impl", + "@local_xla//xla/service/gpu:gpu_symbol_repository", + "@local_xla//xla/service/gpu/autotuning:autotuner_util", + "@local_xla//xla/stream_executor:device_description_proto_cc", + "@local_xla//xla/tests:hlo_test_base", + "@local_xla//xla/tests:xla_internal_test_main", # fixdeps: keep + "@local_xla//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:env", "@local_tsl//tsl/platform:path", @@ -910,9 +910,9 @@ xla_test( srcs = ["hlo_decomposer_test.cc"], deps = [ ":hlo_decomposer_lib", - "//xla/hlo/ir:hlo", - "//xla/tests:filecheck", - "//xla/tests:hlo_test_base", + "@local_xla//xla/hlo/ir:hlo", + "@local_xla//xla/tests:filecheck", + "@local_xla//xla/tests:hlo_test_base", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", diff --git a/third_party/xla/xla/xla.bzl b/third_party/xla/xla/xla.bzl index a81deeb562893c..5e2bcb23a9cea0 100644 --- a/third_party/xla/xla/xla.bzl +++ b/third_party/xla/xla/xla.bzl @@ -15,7 +15,7 @@ load( "if_cuda_is_configured", ) load( - "//xla/tsl:tsl.bzl", + "@local_xla//xla/tsl:tsl.bzl", "tsl_copts", ) @@ -33,36 +33,36 @@ def xla_py_test_deps(): # away. The problem is making sure that all these impl deps are `if_static`'d # appropriately throughout XLA. _XLA_SHARED_OBJECT_SENSITIVE_DEPS = if_static(extra_deps = [], otherwise = [ - Label("//xla:autotune_results_proto_cc_impl"), - Label("//xla:autotuning_proto_cc_impl"), - Label("//xla:xla_data_proto_cc_impl"), - Label("//xla:xla_proto_cc_impl"), - Label("//xla/service:buffer_assignment_proto_cc_impl"), - Label("//xla/service:hlo_proto_cc_impl"), - Label("//xla/service/gpu:backend_configs_cc_impl"), - Label("//xla/service/gpu/model:hlo_op_profile_proto_cc_impl"), - Label("//xla/service/memory_space_assignment:memory_space_assignment_proto_cc_impl"), - Label("//xla/stream_executor:device_description_proto_cc_impl"), - Label("//xla/stream_executor:stream_executor_impl"), - Label("//xla/stream_executor/gpu:gpu_init_impl"), + Label("@local_xla//xla:autotune_results_proto_cc_impl"), + Label("@local_xla//xla:autotuning_proto_cc_impl"), + Label("@local_xla//xla:xla_data_proto_cc_impl"), + Label("@local_xla//xla:xla_proto_cc_impl"), + Label("@local_xla//xla/service:buffer_assignment_proto_cc_impl"), + Label("@local_xla//xla/service:hlo_proto_cc_impl"), + Label("@local_xla//xla/service/gpu:backend_configs_cc_impl"), + Label("@local_xla//xla/service/gpu/model:hlo_op_profile_proto_cc_impl"), + Label("@local_xla//xla/service/memory_space_assignment:memory_space_assignment_proto_cc_impl"), + Label("@local_xla//xla/stream_executor:device_description_proto_cc_impl"), + Label("@local_xla//xla/stream_executor:stream_executor_impl"), + Label("@local_xla//xla/stream_executor/gpu:gpu_init_impl"), "@com_google_protobuf//:protobuf", - "//xla/tsl/framework:allocator_registry_impl", - "//xla/tsl/framework:allocator", + "@local_xla//xla/tsl/framework:allocator_registry_impl", + "@local_xla//xla/tsl/framework:allocator", "@local_tsl//tsl/platform:env_impl", - "//xla/tsl/profiler/backends/cpu:annotation_stack_impl", - "//xla/tsl/profiler/backends/cpu:traceme_recorder_impl", + "@local_xla//xla/tsl/profiler/backends/cpu:annotation_stack_impl", + "@local_xla//xla/tsl/profiler/backends/cpu:traceme_recorder_impl", "@local_tsl//tsl/profiler/protobuf:profiler_options_proto_cc_impl", "@local_tsl//tsl/profiler/protobuf:xplane_proto_cc_impl", "@local_tsl//tsl/profiler/utils:time_utils_impl", "@local_tsl//tsl/protobuf:protos_all_cc_impl", ]) + if_cuda_is_configured([ - Label("//xla/stream_executor/cuda:all_runtime"), - Label("//xla/stream_executor/cuda:stream_executor_cuda"), + Label("@local_xla//xla/stream_executor/cuda:all_runtime"), + Label("@local_xla//xla/stream_executor/cuda:stream_executor_cuda"), ]) + if_rocm_is_configured([ - Label("//xla/stream_executor/gpu:gpu_stream"), - Label("//xla/stream_executor/rocm:all_runtime"), - Label("//xla/stream_executor/rocm:stream_executor_rocm"), - "//xla/tsl/util:determinism", + Label("@local_xla//xla/stream_executor/gpu:gpu_stream"), + Label("@local_xla//xla/stream_executor/rocm:all_runtime"), + Label("@local_xla//xla/stream_executor/rocm:stream_executor_rocm"), + "@local_xla//xla/tsl/util:determinism", ]) def xla_cc_binary(deps = [], copts = tsl_copts(), **kwargs): @@ -88,7 +88,7 @@ def xla_bzl_library(name = "xla_bzl_library"): name = "xla_bzl", srcs = ["xla.bzl"], deps = [ - "//xla/tsl:tsl_bzl", + "@local_xla//xla/tsl:tsl_bzl", "@local_config_rocm//rocm:build_defs_bzl", "@local_tsl//tsl/platform:build_config_root_bzl", "@local_tsl//tsl/platform/default:cuda_build_defs_bzl",