From 97bf2d2bb88d697fd4f888a2002f8a90abe130ff Mon Sep 17 00:00:00 2001 From: Ruturaj4 Date: Mon, 8 Apr 2024 19:58:41 -0500 Subject: [PATCH] [ROCm]: fix tsl path --- jaxlib/rocm/BUILD.bazel | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/jaxlib/rocm/BUILD.bazel b/jaxlib/rocm/BUILD.bazel index ccb66bbc8..c96ab6459 100644 --- a/jaxlib/rocm/BUILD.bazel +++ b/jaxlib/rocm/BUILD.bazel @@ -100,7 +100,7 @@ pybind_extension( "@local_config_rocm//rocm:hipblas", "@local_config_rocm//rocm:rocm_headers", "@nanobind", - "@tsl//tsl/python/lib/core:numpy", + "@xla//xla/tsl/python/lib/core:numpy", ], ) @@ -141,7 +141,7 @@ pybind_extension( "@local_config_rocm//rocm:hipsolver", "@local_config_rocm//rocm:rocm_headers", "@nanobind", - "@tsl//tsl/python/lib/core:numpy", + "@xla//xla/tsl/python/lib/core:numpy", ], ) @@ -189,7 +189,7 @@ pybind_extension( "@local_config_rocm//rocm:hipsparse", "@local_config_rocm//rocm:rocm_headers", "@nanobind", - "@tsl//tsl/python/lib/core:numpy", + "@xla//xla/tsl/python/lib/core:numpy", ], ) @@ -306,7 +306,7 @@ cc_library( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/synchronization", - "@tsl//tsl/platform:env", + "@xla//xla/tsl/util:env_var", ], )