frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] enable HIP IPC

Browse Source 
Enables code paths that use hipIpc* functions.  Also enables test_multiprocessing.py.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74383
Approved by: https://github.com/osalpekar
This commit is contained in:
Jeff Daily
2022-03-21 19:32:01 +00:00
committed by PyTorch MergeBot
parent be1eac0803
commit 956a028b55
4 changed files with 3 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_multiprocessing.py
View File

@ -258,7 +258,7 @@ class TestMultiprocessing(TestCase):
self.assertTrue(e.is_set())
self.assertTrue(data[0].eq(4).all())
self.assertTrue(data[1].eq(4).all())
p.join(1)
p.join(100)
self.assertFalse(p.is_alive())
def test_receive():
@ -280,7 +280,7 @@ class TestMultiprocessing(TestCase):
# collect them properly
del t1, t2
e.set()
p.join(1)
p.join(100)
self.assertFalse(p.is_alive())
with leak_checker(self) as lc:
@ -753,7 +753,7 @@ if __name__ == "__main__":
self.assertEqual(var.data, torch.ones(5, 5, device=device))
self.assertEqual(var.grad.data, torch.ones(5, 5, device=device) * 4)
p.join(1)
p.join(100)
self.assertFalse(p.is_alive())
# Check sharing a cudaMalloc allocation with different types of storage.