[ROCm] [FakeTensorTest] Enable test_fallback_memory_prop (pytorch#85760)

Signed-off-by: Jagadish Krishnamoorthy <jagdish.krishna@gmail.com> Pull Request resolved: pytorch#85760 Approved by: https://github.com/kit1980
lcy-seso · Oct 25, 2022 · 9efca7c · 9efca7c
1 parent e818574
commit 9efca7c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/test/test_fake_tensor.py b/test/test_fake_tensor.py
@@ -329,7 +329,6 @@ def fn(
                         self.assertTrue(isinstance(ten, FakeTensor))
                     self.assertEqual(ten.device.type, 'cuda')
 
-    @skipIfRocm
     @unittest.skipIf(not RUN_CUDA, "requires cuda")
     def test_fallback_memory_prop(self):
         m = nn.Conv2d(16, 33, 3, stride=2, device="cuda", dtype=torch.half)