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

Increase test shard_count for shape_poly_test on GPU

Browse Source 
PiperOrigin-RevId: 723940915
This commit is contained in:
George Necula 2025-02-06 08:13:12 -08:00 committed by jax authors
parent 026b6c9704
commit a678396f44
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

@ -1518,7 +1518,7 @@ jax_multiplatform_test(
],
shard_count = {
"cpu": 4,
"gpu": 4,
"gpu": 6,
"tpu": 4,
},
tags = [