mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm]: fix tsl path

Browse Source
This commit is contained in:
Ruturaj4 2024-04-08 19:58:41 -05:00
parent fed7efd730
commit 97bf2d2bb8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
jaxlib/rocm/BUILD.bazel
View File

@ -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",
],
)