dgl.multiprocessing

这是Python原生multiprocessing模块的最小封装。 它修改了multiprocessing.Process类,以使分叉 在DGL核心库中与OpenMP一起工作。

API 的使用方式与原生模块完全相同,因此 DGL 不提供额外的文档。

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

call_once_and_share(func, shape, dtype[, rank])

在PyTorch分布式进程组的单个进程中调用函数,并与其他进程共享结果。

shared_tensor(形状[, 数据类型])

在共享内存中创建一个张量,该张量可以被同一torch.distributed进程组中的所有进程访问。