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 PgleTest.testAutoPgleWithPersistentCache.

Browse Source 
This is a new test that is failing in CI.

PiperOrigin-RevId: 638619001
This commit is contained in:
Peter Hawkins 2024-05-30 05:32:59 -07:00 committed by jax authors
parent 01b4cb6de0
commit f24d2a71bb
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pgle_test.py
View File

@ -18,6 +18,7 @@ import logging
import math
import os
import tempfile
import unittest
from absl.testing import absltest
import jax
@ -149,6 +150,7 @@ class PgleTest(jtu.JaxTestCase):
self.assertArraysEqual(compiled(x), expected)
self.assertEqual(cache_miss_count[0], 0)
@unittest.skip("Test failing in CI")
def testAutoPgleWithPersistentCache(self):
if xla_extension_version < 268:
return self.skipTest('Requires xla_extension_version >= 268')