dgl.multiprocessing

这是 Python 原生 multiprocessing 模块的一个最小包装。它修改了 multiprocessing.Process 类，以使 DGL 核心库中的 OpenMP 能够使用 fork。

API 用法与原生模块完全相同，因此 DGL 不提供额外的文档。

此外，如果您的后端是 PyTorch，此模块也将与 torch.multiprocessing 模块兼容。

call_once_and_share(func, shape, dtype[, rank])	在 PyTorch 分布式进程组的单个进程中调用函数，并将结果与其他进程共享。
shared_tensor(shape[, dtype])	在共享内存中创建一个张量，该张量可由同一 torch.distributed 进程组内的所有进程访问。