Skip to content

Commit 9431ba9

Browse files
committed
Skip tests in test_torch that got enabled through the LAPACK addition
and are failing.
1 parent 6868c98 commit 9431ba9

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/test_torch.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3931,6 +3931,7 @@ def test_is_signed_cuda(self):
39313931
self.assertEqual(torch.cuda.HalfTensor(10).is_signed(), True)
39323932

39333933
@skipIfNoLapack
3934+
@skipIfRocm
39343935
def test_gesv(self):
39353936
a = torch.Tensor(((6.80, -2.11, 5.66, 5.97, 8.23),
39363937
(-6.05, -3.30, 5.36, -4.44, 1.08),
@@ -4061,6 +4062,7 @@ def test_gesv_batched_dims(self):
40614062
self._test_gesv_batched_dims(self, lambda t: t)
40624063

40634064
@skipIfNoLapack
4065+
@skipIfRocm
40644066
def test_qr(self):
40654067

40664068
# Since the QR decomposition is unique only up to the signs of the rows of
@@ -4243,10 +4245,12 @@ def _test_trtrs(self, cast):
42434245
self.assertEqual(res1, tb, 0)
42444246

42454247
@skipIfNoLapack
4248+
@skipIfRocm
42464249
def test_trtrs(self):
42474250
self._test_trtrs(self, lambda t: t)
42484251

42494252
@skipIfNoLapack
4253+
@skipIfRocm
42504254
def test_gels(self):
42514255
def _test_underdetermined(a, b, expectedNorm):
42524256
m = a.size()[0]
@@ -4362,6 +4366,7 @@ def check_norm(a, b, expected_norm, gels_result):
43624366
self.assertEqual((torch.mm(a, tb) - b).norm(), expectedNorm, 1e-8)
43634367

43644368
@skipIfNoLapack
4369+
@skipIfRocm
43654370
def test_eig(self):
43664371
a = torch.Tensor(((1.96, 0.00, 0.00, 0.00, 0.00),
43674372
(-6.49, 3.80, 0.00, 0.00, 0.00),

0 commit comments

Comments
 (0)