Skip to content

Commit 34b0864

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 98c5cd8 commit 34b0864

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
@@ -1907,6 +1907,7 @@ def test_gamma_sample(self):
19071907

19081908
@unittest.skipIf(not TEST_CUDA, "CUDA not found")
19091909
@unittest.skipIf(not TEST_NUMPY, "Numpy not found")
1910+
@skipIfRocm
19101911
def test_gamma_gpu_sample(self):
19111912
set_rng_seed(0)
19121913
for alpha, beta in product([0.1, 1.0, 5.0], [0.1, 1.0, 10.0]):
@@ -3984,6 +3985,7 @@ def test_transform_to(self):
39843985
self.assertEqual(y, y2, message="Error in transform_to({}) pseudoinverse".format(constraint))
39853986

39863987
@unittest.skipIf(not TEST_CUDA, "CUDA not found")
3988+
@skipIfRocm
39873989
def test_transform_to_cuda(self):
39883990
for constraint in self.get_constraints(is_cuda=True):
39893991
t = transform_to(constraint)

0 commit comments

Comments
 (0)