项目文件夹

文件
Quan (Andy) Gan caa6d6072c [Feature] Replacing thread_wrapped_func with minimal mp.Process wrapper (#2905)
* standardizing thread_wrapped_func

* lints

* Update __init__.py
2021-05-14 11:34:49 +08:00

15 行
482 B
ReStructuredText

.. _apimultiprocessing:
dgl.multiprocessing
===================
This is a minimal wrapper of Python's native :mod:`multiprocessing` module.
It modifies the :class:`multiprocessing.Process` class to make forking
work with OpenMP in the DGL core library.
The API usage is exactly the same as the native module, so DGL does not provide
additional documentation.
In addition, if your backend is PyTorch, this module will also be compatible with
:mod:`torch.multiprocessing` module.