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

Disable idct tests on GPU.

Browse Source 
We recently switched our CI to use T4 GPUs instead of P100s, which causes these tests to fail. Disable the tests for now.

PiperOrigin-RevId: 551910196
This commit is contained in:
Peter Hawkins 2023-07-28 11:45:55 -07:00 committed by jax authors
parent 19eefac1e8
commit 7f30ae3f00
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/scipy_fft_test.py
View File

@ -86,6 +86,9 @@ class LaxBackedScipyFftTests(jtu.JaxTestCase):
axis=[-1, 0],
norm=[None, 'ortho'],
)
# TODO(phawkins): these tests are failing on T4 GPUs in CI with a
# CUDA_ERROR_ILLEGAL_ADDRESS.
@jtu.skip_on_devices("cuda")
def testiDct(self, shape, dtype, n, axis, norm):
rng = jtu.rand_default(self.rng())
args_maker = lambda: (rng(shape, dtype),)