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

Fix ROCM BUILD rule

Browse Source 
Fixes https://github.com/google/jax/issues/9864; replaces https://github.com/google/jax/issues/9870

PiperOrigin-RevId: 434554684
This commit is contained in:
Jake VanderPlas 2022-03-14 13:38:53 -07:00 committed by jax authors
parent 76d340f0ac
commit 765d11d50c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/BUILD
View File

@ -107,7 +107,7 @@ cc_library(
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
] + if_rocm_is_configured([
"@org_tensorflow//tensorflow/third_party/rocm/google:rocm_headers",
"@local_config_rocm//rocm:rocm_headers",
]),
)