Kiuk Chung abd16a8c64 [torch/multiprocessing] Use multiprocessing.reduction.register ForkingPickler.register to register custom tensor and storage reductions (#135030) ... Right now `multiprocessing.reduction.register()` is simply an alias to `multiprocessing.reduction.ForkingPickler.register()` https://github.com/python/cpython/blame/main/Lib/multiprocessing/reduction.py#L56, but the top-level `register()` function exposes less of the internal details of `multiprocessing.reduction` module. Pull Request resolved: https://github.com/pytorch/pytorch/pull/135030 Approved by: https://github.com/albanD		2024-09-16 20:07:29 +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	Fix docstring errors in reductions.py, spawn.py, pool.py, parameter.py, cpp.py, grad.py, __init__.py, profiler.py, queue.py, graph.py (#113052)	2023-11-10 21:19:17 +00:00
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	Create processes in parallel in mp.start_processes for forkserver (#134629)	2024-08-28 21:34:32 +00:00