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

Enable test_scan_offload in memories_test.

Browse Source 
PiperOrigin-RevId: 742840628
This commit is contained in:
jax authors 2025-04-01 14:25:40 -07:00
parent 2eb5c8a6ea
commit ff5a2e8c91
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/memories_test.py
View File

@ -1090,7 +1090,6 @@ class ComputeOffload(jtu.BufferDonationTestCase):
self.assertArraysEqual(g(arr), all_true)
def test_scan_offload(self):
self.skipTest('b/406586554')
np_inp = jnp.arange(4096).reshape(16, 16, 16)
@jax.jit