mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
5976f0bdfe
Also, grep for `if sys.version_info .cond. (3, 8)` and replaces them with appropriate action. This is a last in a series of PRs that moved CI/CD away from testing PyTorch behavior against Python-3.7. Fixes https://github.com/pytorch/pytorch/issues/80513 Pull Request resolved: https://github.com/pytorch/pytorch/pull/93155 Approved by: https://github.com/huydhn
31 lines
759 B
Python
31 lines
759 B
Python
import sys
|
|
|
|
__all__ = ['register_after_fork']
|
|
|
|
if sys.platform == 'win32':
|
|
import multiprocessing.util as _util
|
|
|
|
def _register(func):
|
|
def wrapper(arg):
|
|
func()
|
|
_util.register_after_fork(_register, wrapper)
|
|
else:
|
|
import os
|
|
|
|
def _register(func):
|
|
os.register_at_fork(after_in_child=func)
|
|
|
|
def register_after_fork(func):
|
|
"""Register a callable to be executed in the child process after a fork.
|
|
|
|
Note:
|
|
In python < 3.7 this will only work with processes created using the
|
|
``multiprocessing`` module. In python >= 3.7 it also works with
|
|
``os.fork()``.
|
|
|
|
Args:
|
|
func (function): Function taking no arguments to be called in the child after fork
|
|
|
|
"""
|
|
_register(func)
|