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

Reenable scipy_stats_test in CI.

Browse Source 
Disable testTruncNormPdf on CPU, which is failing after an LLVM update.

PiperOrigin-RevId: 528884880
This commit is contained in:
Peter Hawkins 2023-05-02 14:10:30 -07:00 committed by jax authors
parent 40349a8612
commit 57e62ca03c
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/BUILD
View File

@ -727,7 +727,6 @@ jax_test(
},
tags = [
"noasan",
"notap", # TODO(b/271883906): Failing on xla.opt.debug.tpu_hw tap
"notsan",
], # Times out
)

4
tests/scipy_stats_test.py
View File

@ -12,9 +12,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.
from functools import partial
import itertools
import unittest
from absl.testing import absltest
@ -767,6 +767,8 @@ class LaxBackedScipyStatsTests(jtu.JaxTestCase):
@genNamedParametersNArgs(5)
def testTruncnormPdf(self, shapes, dtypes):
if jtu.device_under_test() == "cpu":
raise unittest.SkipTest("TODO(b/280418378) test fails after LLVM update")
rng = jtu.rand_default(self.rng())
scipy_fun = osp_stats.truncnorm.pdf
lax_fun = lsp_stats.truncnorm.pdf