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

This reverts commit cbb2d5af81dcfaf181db7e9083b9c41b29fdb4eb.

Reverted https://github.com/pytorch/pytorch/pull/90335 on behalf of https://github.com/clee2000 due to somehow caused test_multiprocessing to timeout cbb2d5af81 https://github.com/pytorch/pytorch/actions/runs/3645873711/jobs/6159998523
This commit is contained in:
PyTorch MergeBot
2022-12-08 17:12:10 +00:00
parent 8ea90d926f
commit 465005c1e0

View File

@ -418,7 +418,8 @@ class TestMultiprocessing(TestCase):
t = []
for _ in range(5):
t.append(q.get())
self.assertEqual(t[0], torch.full([5], 0.))
# TODO(#38095): Replace assertEqualIgnoreType. See issue #38095
self.assertEqualIgnoreType(t[0], torch.full([5], 0.))
del t
e.set()
p.join(1)
@ -641,7 +642,7 @@ if __name__ == "__main__":
c2p.put(0) # notify parent child is ready
p2c.get() # wait for record in parent
e1.synchronize()
c2p.put(1) # notify synchronization is done in child
c2p.put(1) # nofity 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 \