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

Increase shard count after adding more tests

Browse Source 
PiperOrigin-RevId: 705146601
This commit is contained in:
Gleb Pobudzey 2024-12-11 10:08:08 -08:00 committed by jax authors
parent 01206f839b
commit 20236f1083
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/pallas/BUILD
View File

@ -481,7 +481,7 @@ jax_multiplatform_test(
"gpu_a100_x32",
"gpu_h100_x32",
],
shard_count = 2,
shard_count = 6,
deps = [
"//jax:pallas",
"//jax:pallas_gpu",