mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

CI changes

Browse Source 
PiperOrigin-RevId: 563754696
This commit is contained in:
Adam Paszke 2023-09-08 08:07:30 -07:00 committed by jax authors
parent 592eb44d7e
commit 832820e8ff
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/pallas/BUILD
View File

@ -31,6 +31,11 @@ jax_test(
srcs = [
"pallas_test.py",
],
config_tags_overrides = {
"gpu_x32": {
"ondemand": False, # Include in presubmit.
},
},
disable_backends = [
"cpu",
"tpu",