OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: Jagadish Krishnamoorthy <jagdish.krishna@gmail.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/85760
Approved by: https://github.com/kit1980
This commit is contained in:
Jagadish Krishnamoorthy 2022-10-25 07:17:44 +00:00 committed by PyTorch MergeBot
parent e818574e78
commit 9efca7c085
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_fake_tensor.py
View File

@ -329,7 +329,6 @@ class FakeTensorTest(TestCase):
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)