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

Internal change

Browse Source 
PiperOrigin-RevId: 484330913
This commit is contained in:
Jake VanderPlas 2022-10-27 13:11:58 -07:00 committed by jax authors
parent 3e38675ac4
commit f699acd931
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/BUILD
View File

@ -700,6 +700,7 @@ jax_test(
srcs = ["sparsify_test.py"],
args = ["--jax_bcoo_cusparse_lowering=true"],
shard_count = {
"cpu": 5,
"gpu": 20,
"tpu": 10,
},