diff --git a/examples/jax_cpp/BUILD b/examples/jax_cpp/BUILD index d6646a84a..88cf83848 100644 --- a/examples/jax_cpp/BUILD +++ b/examples/jax_cpp/BUILD @@ -19,15 +19,15 @@ cc_binary( srcs = ["main.cc"], tags = ["manual"], deps = [ - "@xla//:literal", - "@xla//:literal_util", - "@xla//:shape_util", - "@xla//:status", - "@xla//:statusor", - "@xla///pjrt:pjrt_client", - "@xla///pjrt:tfrt_cpu_pjrt_client", - "@xla///service:hlo_proto_cc", - "@xla///tools:hlo_module_loader", + "@xla//xla:literal", + "@xla//xla:literal_util", + "@xla//xla:shape_util", + "@xla//xla:status", + "@xla//xla:statusor", + "@xla//xla/pjrt:pjrt_client", + "@xla//xla/pjrt:tfrt_cpu_pjrt_client", + "@xla//xla/service:hlo_proto_cc", + "@xla//xla/tools:hlo_module_loader", "@tsl///platform:logging", "@tsl///platform:platform_port", ], diff --git a/jaxlib/BUILD b/jaxlib/BUILD index bbb5cf574..ffe92525e 100644 --- a/jaxlib/BUILD +++ b/jaxlib/BUILD @@ -68,7 +68,7 @@ symlink_files( symlink_files( name = "xla_client", - srcs = ["@xla///python:xla_client"], + srcs = ["@xla//xla/python:xla_client"], dst = ".", flatten = True, ) @@ -76,8 +76,8 @@ symlink_files( symlink_files( name = "xla_extension", srcs = if_windows( - ["@xla///python:xla_extension.pyd"], - ["@xla///python:xla_extension.so"], + ["@xla//xla/python:xla_extension.pyd"], + ["@xla//xla/python:xla_extension.so"], ), dst = ".", flatten = True, diff --git a/jaxlib/cpu/BUILD b/jaxlib/cpu/BUILD index b3f9ca10b..85af47c9b 100644 --- a/jaxlib/cpu/BUILD +++ b/jaxlib/cpu/BUILD @@ -31,7 +31,7 @@ cc_library( srcs = ["lapack_kernels.cc"], hdrs = ["lapack_kernels.h"], deps = [ - "@xla///service:custom_call_status", + "@xla//xla/service:custom_call_status", "@com_google_absl//absl/base:dynamic_annotations", ], ) @@ -74,7 +74,7 @@ cc_library( features = ["-use_header_modules"], deps = [ ":ducc_fft_flatbuffers_cc", - "@xla///service:custom_call_status", + "@xla//xla/service:custom_call_status", "@ducc", "@flatbuffers//:runtime_cc", ], @@ -106,7 +106,7 @@ cc_library( ":ducc_fft_kernels", ":lapack_kernels", ":lapack_kernels_using_lapack", - "@xla///service:custom_call_target_registry", + "@xla//xla/service:custom_call_target_registry", ], alwayslink = 1, ) diff --git a/jaxlib/cuda/BUILD b/jaxlib/cuda/BUILD index 94f964050..3943b603d 100644 --- a/jaxlib/cuda/BUILD +++ b/jaxlib/cuda/BUILD @@ -51,8 +51,8 @@ cc_library( features = ["-use_header_modules"], deps = [ ":cuda_vendor", - "@xla///stream_executor/cuda:cusolver_lib", - "@xla///stream_executor/cuda:cusparse_lib", + "@xla//xla/stream_executor/cuda:cusolver_lib", + "@xla//xla/stream_executor/cuda:cusparse_lib", "@com_google_absl//absl/memory", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", @@ -72,9 +72,9 @@ cc_library( ":cuda_vendor", "//jaxlib:handle_pool", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", - "@xla///stream_executor/cuda:cublas_lib", - "@xla///stream_executor/cuda:cudart_stub", + "@xla//xla/service:custom_call_status", + "@xla//xla/stream_executor/cuda:cublas_lib", + "@xla//xla/stream_executor/cuda:cudart_stub", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/base", "@com_google_absl//absl/base:core_headers", @@ -102,7 +102,7 @@ pybind_extension( ":cublas_kernels", ":cuda_vendor", "//jaxlib:kernel_pybind11_helpers", - "@xla///stream_executor/cuda:cublas_lib", + "@xla//xla/stream_executor/cuda:cublas_lib", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/strings:str_format", "@pybind11", @@ -118,9 +118,9 @@ cc_library( ":cuda_vendor", "//jaxlib:handle_pool", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", - "@xla///stream_executor/cuda:cudart_stub", - "@xla///stream_executor/cuda:cudnn_lib", + "@xla//xla/service:custom_call_status", + "@xla//xla/stream_executor/cuda:cudart_stub", + "@xla//xla/stream_executor/cuda:cudnn_lib", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:str_format", @@ -157,8 +157,8 @@ cc_library( ":cuda_vendor", "//jaxlib:handle_pool", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", - "@xla///stream_executor/cuda:cusolver_lib", + "@xla//xla/service:custom_call_status", + "@xla//xla/stream_executor/cuda:cusolver_lib", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/synchronization", @@ -180,8 +180,8 @@ pybind_extension( ":cuda_vendor", ":cusolver_kernels", "//jaxlib:kernel_pybind11_helpers", - "@xla///stream_executor/cuda:cudart_stub", - "@xla///stream_executor/cuda:cusolver_lib", + "@xla//xla/stream_executor/cuda:cudart_stub", + "@xla//xla/stream_executor/cuda:cusolver_lib", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/strings:str_format", "@local_config_cuda//cuda:cuda_headers", @@ -198,9 +198,9 @@ cc_library( ":cuda_vendor", "//jaxlib:handle_pool", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", - "@xla///stream_executor/cuda:cudart_stub", - "@xla///stream_executor/cuda:cusparse_lib", + "@xla//xla/service:custom_call_status", + "@xla//xla/stream_executor/cuda:cudart_stub", + "@xla//xla/stream_executor/cuda:cusparse_lib", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/synchronization", @@ -222,8 +222,8 @@ pybind_extension( ":cuda_vendor", ":cusparse_kernels", "//jaxlib:kernel_pybind11_helpers", - "@xla///stream_executor/cuda:cudart_stub", - "@xla///stream_executor/cuda:cusparse_lib", + "@xla//xla/stream_executor/cuda:cudart_stub", + "@xla//xla/stream_executor/cuda:cusparse_lib", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/base", "@com_google_absl//absl/base:core_headers", @@ -249,7 +249,7 @@ cc_library( ":cuda_lu_pivot_kernels_impl", ":cuda_vendor", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", + "@xla//xla/service:custom_call_status", "@local_config_cuda//cuda:cuda_headers", ], ) @@ -264,7 +264,7 @@ cuda_library( ":cuda_gpu_kernel_helpers", ":cuda_vendor", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", + "@xla//xla/service:custom_call_status", "@local_config_cuda//cuda:cuda_headers", ], ) @@ -284,7 +284,7 @@ pybind_extension( ":cuda_lu_pivot_kernels_impl", ":cuda_vendor", "//jaxlib:kernel_pybind11_helpers", - "@xla///stream_executor/cuda:cudart_stub", + "@xla//xla/stream_executor/cuda:cudart_stub", "@local_config_cuda//cuda:cuda_headers", "@pybind11", ], @@ -301,7 +301,7 @@ cc_library( ":cuda_prng_kernels_impl", ":cuda_vendor", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", + "@xla//xla/service:custom_call_status", "@local_config_cuda//cuda:cuda_headers", ], ) @@ -316,7 +316,7 @@ cuda_library( ":cuda_gpu_kernel_helpers", ":cuda_vendor", "//jaxlib:kernel_helpers", - "@xla///service:custom_call_status", + "@xla//xla/service:custom_call_status", "@local_config_cuda//cuda:cuda_headers", ], ) @@ -334,7 +334,7 @@ pybind_extension( ":cuda_gpu_kernel_helpers", ":cuda_prng_kernels", "//jaxlib:kernel_pybind11_helpers", - "@xla///stream_executor/cuda:cudart_stub", + "@xla//xla/stream_executor/cuda:cudart_stub", "@local_config_cuda//cuda:cuda_headers", "@pybind11", ], @@ -351,7 +351,7 @@ cc_library( ":cuda_vendor", ":cusolver_kernels", ":cusparse_kernels", - "@xla///service:custom_call_target_registry", + "@xla//xla/service:custom_call_target_registry", ], alwayslink = 1, )