diff --git a/test/test_torch.py b/test/test_torch.py index ea0d69d7b0e265..dccd13474ad23d 100644 --- a/test/test_torch.py +++ b/test/test_torch.py @@ -3931,6 +3931,7 @@ def test_is_signed_cuda(self): self.assertEqual(torch.cuda.HalfTensor(10).is_signed(), True) @skipIfNoLapack + @skipIfRocm def test_gesv(self): a = torch.Tensor(((6.80, -2.11, 5.66, 5.97, 8.23), (-6.05, -3.30, 5.36, -4.44, 1.08), @@ -4061,6 +4062,7 @@ def test_gesv_batched_dims(self): self._test_gesv_batched_dims(self, lambda t: t) @skipIfNoLapack + @skipIfRocm def test_qr(self): # Since the QR decomposition is unique only up to the signs of the rows of @@ -4243,10 +4245,12 @@ def _test_trtrs(self, cast): self.assertEqual(res1, tb, 0) @skipIfNoLapack + @skipIfRocm def test_trtrs(self): self._test_trtrs(self, lambda t: t) @skipIfNoLapack + @skipIfRocm def test_gels(self): def _test_underdetermined(a, b, expectedNorm): m = a.size()[0] @@ -4362,6 +4366,7 @@ def check_norm(a, b, expected_norm, gels_result): self.assertEqual((torch.mm(a, tb) - b).norm(), expectedNorm, 1e-8) @skipIfNoLapack + @skipIfRocm def test_eig(self): a = torch.Tensor(((1.96, 0.00, 0.00, 0.00, 0.00), (-6.49, 3.80, 0.00, 0.00, 0.00),