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

Fix issue 38095 TODO in test_multiprocessing.py (#90335)

Browse Source 
Fix TODO related to https://github.com/pytorch/pytorch/issues/38095
Pull Request resolved: https://github.com/pytorch/pytorch/pull/90335
Approved by: https://github.com/clee2000
This commit is contained in:
Sergii Dymchenko
2022-12-08 06:27:08 +00:00
committed by PyTorch MergeBot
parent 06c98e673f
commit cbb2d5af81
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_multiprocessing.py
View File

@ -418,8 +418,7 @@ class TestMultiprocessing(TestCase):
t = []
for _ in range(5):
t.append(q.get())
# TODO(#38095): Replace assertEqualIgnoreType. See issue #38095
self.assertEqualIgnoreType(t[0], torch.full([5], 0.))
self.assertEqual(t[0], torch.full([5], 0.))
del t
e.set()
p.join(1)
@ -642,7 +641,7 @@ if __name__ == "__main__":
c2p.put(0) # notify parent child is ready
p2c.get() # wait for record in parent
e1.synchronize()
c2p.put(1) # nofity synchronization is done in child
c2p.put(1) # notify synchronization is done in child
p2c.get() # wait for parent to finish before destructing child event
@unittest.skipIf(NO_MULTIPROCESSING_SPAWN, "Disabled for environments that \