[WIP] add tests for loop-based gesvd
@@ -257,6 +257,11 @@ class TestSVD(unittest.TestCase):
self.check_usv((2, 7, 3))
self.check_usv((2, 4, 3))
+ @condition.repeat(3, 10)
+ def test_svd_rank3_loop(self):
+ # This tests the loop-based batched gesvd on CUDA
+ self.check_usv((2, 64, 64))
+
@condition.repeat(3, 10)
def test_svd_rank3_no_uv(self):
self.check_singular((2, 3, 4))
@@ -265,6 +270,11 @@ class TestSVD(unittest.TestCase):
self.check_singular((2, 7, 3))
self.check_singular((2, 4, 3))
+ @condition.repeat(3, 10)
+ def test_svd_rank3_no_uv_loop(self):
+ # This tests the loop-based batched gesvd on CUDA
+ self.check_singular((2, 64, 64))
+
@testing.with_requires('numpy>=1.16')
def test_svd_rank3_empty_array(self):
self.check_usv((0, 3, 4))
@@ -296,6 +306,11 @@ class TestSVD(unittest.TestCase):
self.check_usv((5, 2, 7, 3))
self.check_usv((5, 2, 4, 3))
+ @condition.repeat(3, 10)
+ def test_svd_rank4_loop(self):
+ # This tests the loop-based batched gesvd on CUDA
+ self.check_usv((3, 2, 64, 64))
+
@condition.repeat(3, 10)
def test_svd_rank4_no_uv(self):
self.check_singular((5, 2, 3, 4))
@@ -304,6 +319,10 @@ class TestSVD(unittest.TestCase):
self.check_singular((5, 2, 7, 3))
self.check_singular((5, 2, 4, 3))
+ @condition.repeat(3, 10)
+ def test_svd_rank4_no_uv_loop(self):
+ self.check_singular((3, 2, 64, 64))
+
@testing.with_requires('numpy>=1.16')
def test_svd_rank4_empty_array(self):
self.check_usv((0, 2, 3, 4))
