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

fix building on ROCm

Browse Source
This commit is contained in:
Clemens Giuliani 2021-01-20 10:49:29 +01:00
parent 619bfd549a
commit 57405b0dc1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/BUILD
View File

@ -244,6 +244,8 @@ pybind_extension(
"@com_google_absl//absl/strings:str_format",
"@com_google_absl//absl/synchronization",
"@local_config_rocm//rocm:rocm_headers",
"@local_config_rocm//rocm:rocblas",
"@local_config_rocm//rocm:rocsolver",
"@pybind11",
],
)