mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
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:
@ -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 \
|
||||
|
Reference in New Issue
Block a user