[ROCm] Skip test_masked_scatter_large_tensor_cuda

Refer #60190 Signed-off-by: Jagadish Krishnamoorthy <jagdish.krishna@gmail.com>
pytorch · Jul 6, 2021 · 4a86916 · 4a86916
1 parent 45cc207
commit 4a86916
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -5911,6 +5911,8 @@ def test_masked_scatter_bool_tensor(self, device):
         dst = dst.masked_scatter(mask, src)
         self.assertEqual(dst, torch.tensor([True, True, True], device=device))
 
+    # refer https://github.com/pytorch/pytorch/issues/60190
+    @skipIfRocm
     @onlyCUDA
     @largeTensorTest('30GB')
     def test_masked_scatter_large_tensor(self, device):