diff --git a/test/test_sparse.py b/test/test_sparse.py index 831f0f746ae312..a91681d4767049 100644 --- a/test/test_sparse.py +++ b/test/test_sparse.py @@ -496,6 +496,7 @@ def test_shape(sparse_dims, nnz, with_size): test_shape(3, 10, [100, 100, 100, 5, 5, 5, 0]) test_shape(3, 0, [0, 0, 100, 5, 5, 5, 0]) + @skipIfRocm def test_Sparse_to_Sparse_copy_(self): # This is for testing torch.copy_(SparseTensor, SparseTensor) sparse_dims = 3 @@ -535,6 +536,7 @@ def test_Sparse_to_Sparse_copy_(self): self.assertEqual(None, x1.grad) @unittest.skipIf(torch.cuda.device_count() < 2, "no multi-GPU") + @skipIfRocm def test_Sparse_to_Sparse_copy_multi_gpu(self): # This is for testing torch.copy_(SparseTensor, SparseTensor) across GPU devices sparse_dims = 3