Skip to content

Commit 2cdc205

Browse files
authored
Merge pull request #207 from iotamudelta/sparsefix
Skip tests in test_torch that got enabled through the LAPACK addition
2 parents 3b80f6c + 9431ba9 commit 2cdc205

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
@@ -3932,6 +3932,7 @@ def test_is_signed_cuda(self):
39323932
self.assertEqual(torch.cuda.HalfTensor(10).is_signed(), True)
39333933

39343934
@skipIfNoLapack
3935+
@skipIfRocm
39353936
def test_gesv(self):
39363937
a = torch.Tensor(((6.80, -2.11, 5.66, 5.97, 8.23),
39373938
(-6.05, -3.30, 5.36, -4.44, 1.08),
@@ -4062,6 +4063,7 @@ def test_gesv_batched_dims(self):
40624063
self._test_gesv_batched_dims(self, lambda t: t)
40634064

40644065
@skipIfNoLapack
4066+
@skipIfRocm
40654067
def test_qr(self):
40664068

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

42464248
@skipIfNoLapack
4249+
@skipIfRocm
42474250
def test_trtrs(self):
42484251
self._test_trtrs(self, lambda t: t)
42494252

42504253
@skipIfNoLapack
4254+
@skipIfRocm
42514255
def test_gels(self):
42524256
def _test_underdetermined(a, b, expectedNorm):
42534257
m = a.size()[0]
@@ -4363,6 +4367,7 @@ def check_norm(a, b, expected_norm, gels_result):
43634367
self.assertEqual((torch.mm(a, tb) - b).norm(), expectedNorm, 1e-8)
43644368

43654369
@skipIfNoLapack
4370+
@skipIfRocm
43664371
def test_eig(self):
43674372
a = torch.Tensor(((1.96, 0.00, 0.00, 0.00, 0.00),
43684373
(-6.49, 3.80, 0.00, 0.00, 0.00),

0 commit comments

Comments
 (0)