mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Temporarily disable LaxBackedScipyStatsTests.testTruncnormPdf.

Browse Source 
This test started failing at LLVM head.

PiperOrigin-RevId: 532095958
This commit is contained in:
Peter Hawkins 2023-05-15 06:52:05 -07:00 committed by jax authors
parent 78f6246c2f
commit f7f1ddbb1e
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_stats_test.py
View File

@ -14,6 +14,7 @@
from functools import partial
import itertools
import unittest
from absl.testing import absltest
@ -766,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/282695039): test fails at LLVM head")
rng = jtu.rand_default(self.rng())
scipy_fun = osp_stats.truncnorm.pdf
lax_fun = lsp_stats.truncnorm.pdf