Skip to content

Commit 2647a53

Browse files
committed
Skip two tests that cause a runtime error (wrong type,
test_transform_to_cuda) and a assertion error (wrong result, test_gamma_gpu_sample)
1 parent 7352699 commit 2647a53

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/test_distributions.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1972,6 +1972,7 @@ def test_gamma_sample(self):
19721972

19731973
@unittest.skipIf(not TEST_CUDA, "CUDA not found")
19741974
@unittest.skipIf(not TEST_NUMPY, "Numpy not found")
1975+
@skipIfRocm
19751976
def test_gamma_gpu_sample(self):
19761977
set_rng_seed(0)
19771978
for alpha, beta in product([0.1, 1.0, 5.0], [0.1, 1.0, 10.0]):
@@ -4123,6 +4124,7 @@ def test_transform_to(self):
41234124
self.assertEqual(y, y2, message="Error in transform_to({}) pseudoinverse".format(constraint))
41244125

41254126
@unittest.skipIf(not TEST_CUDA, "CUDA not found")
4127+
@skipIfRocm
41264128
def test_transform_to_cuda(self):
41274129
for constraint in self.get_constraints(is_cuda=True):
41284130
t = transform_to(constraint)

0 commit comments

Comments
 (0)