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 shards for checkify_test on GPU

Browse Source 
PiperOrigin-RevId: 501430172
This commit is contained in:
Sharad Vikram 2023-01-11 18:25:30 -08:00 committed by jax authors
parent f0cca20a67
commit f729da4a36
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/BUILD
View File

@ -779,6 +779,9 @@ jax_test(
name = "checkify_test",
srcs = ["checkify_test.py"],
pjrt_c_api_bypass = True,
shard_count = {
"gpu": 2,
},
)
jax_test(