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

Remove cublas_header by default for anything under jaxlib/*

Browse Source 
PiperOrigin-RevId: 388802894
This commit is contained in:
Yash Katariya 2021-08-04 15:53:08 -07:00 committed by jax authors
parent ab364296cb
commit b5b44b0639
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jaxlib/BUILD
View File

@ -85,6 +85,7 @@ cc_library(
"@com_google_absl//absl/memory",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/strings:str_format",
"@local_config_cuda//cuda:cublas_headers",
"@local_config_cuda//cuda:cuda_headers",
],
)