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

Merge pull request #20646 from ROCm:rcom-ci-tsl-path-fix

Browse Source 
PiperOrigin-RevId: 623129753
This commit is contained in:
jax authors 2024-04-09 05:09:11 -07:00
commit f5cc272615
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",
],
)