TEST: skip CUDA 10.0
@@ -256,6 +256,8 @@ class TestSVD(unittest.TestCase):
self.check_usv((2, 4, 3))
@condition.repeat(3, 10)
+ @pytest.mark.skipif(cupy.cuda.runtime.runtimeGetVersion() == 10000,
+ reason='cuSOLVER bug')
def test_svd_rank3_loop(self):
# This tests the loop-based batched gesvd on CUDA
self.check_usv((2, 64, 64))
@@ -269,6 +271,8 @@ class TestSVD(unittest.TestCase):
self.check_singular((2, 4, 3))
@condition.repeat(3, 10)
+ @pytest.mark.skipif(cupy.cuda.runtime.runtimeGetVersion() == 10000,
+ reason='cuSOLVER bug')
def test_svd_rank3_no_uv_loop(self):
# This tests the loop-based batched gesvd on CUDA
self.check_singular((2, 64, 64))
@@ -305,6 +309,8 @@ class TestSVD(unittest.TestCase):
self.check_usv((2, 2, 4, 3))
@condition.repeat(3, 10)
+ @pytest.mark.skipif(cupy.cuda.runtime.runtimeGetVersion() == 10000,
+ reason='cuSOLVER bug')
def test_svd_rank4_loop(self):
# This tests the loop-based batched gesvd on CUDA
self.check_usv((3, 2, 64, 64))
@@ -318,6 +324,8 @@ class TestSVD(unittest.TestCase):
self.check_singular((2, 2, 4, 3))
@condition.repeat(3, 10)
+ @pytest.mark.skipif(cupy.cuda.runtime.runtimeGetVersion() == 10000,
+ reason='cuSOLVER bug')
def test_svd_rank4_no_uv_loop(self):
self.check_singular((3, 2, 64, 64))
