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

Add more shards to for_loop_test_cpu

Browse Source 
PiperOrigin-RevId: 484309176
This commit is contained in:
Sharad Vikram 2022-10-27 11:41:34 -07:00 committed by jax authors
parent 4b1fd63263
commit 4d94908e6b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/BUILD
View File

@ -935,7 +935,7 @@ jax_test(
name = "for_loop_test",
srcs = ["for_loop_test.py"],
shard_count = {
"cpu": 10,
"cpu": 20,
"gpu": 10,
"tpu": 10,
},