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

Re-enable testTruncNormPdf on CPU

Browse Source 
Breaking change was reverted in LLVM 3b8bc83527

PiperOrigin-RevId: 529072697
This commit is contained in:
Benjamin Kramer 2023-05-03 06:31:24 -07:00 committed by jax authors
parent a2ac510dc3
commit 545c483e50
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/scipy_stats_test.py
View File

@ -14,7 +14,6 @@
from functools import partial
import itertools
import unittest
from absl.testing import absltest
@ -767,8 +766,6 @@ 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