pytorch/torch/multiprocessing
Yuxin Wu 8c3ab21490 multiprocessing.spawn: allow a grace period when shutdown (#131278)
When one process fails, others are immediately killed. This prevents other processes to do necessary cleanups, or dump debug information (in particular, the NCCL flight recorder).

This PR adds a grace period. Default behavior is unchanged.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131278
Approved by: https://github.com/albanD
2024-10-07 12:37:34 +00:00
..
__init__.py Create processes in parallel in mp.start_processes for forkserver (#134629) 2024-08-28 21:34:32 +00:00
_atfork.py [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
cuda_multiprocessing.md
pool.py
queue.py Flip default value for mypy disallow_untyped_defs [7/11] (#127844) 2024-06-08 18:49:45 +00:00
reductions.py [torch/multiprocessing] Use multiprocessing.reduction.register ForkingPickler.register to register custom tensor and storage reductions (#135030) 2024-09-16 20:07:29 +00:00
spawn.py multiprocessing.spawn: allow a grace period when shutdown (#131278) 2024-10-07 12:37:34 +00:00