dgl.multiprocessing
这是Python原生multiprocessing
模块的最小封装。
它修改了multiprocessing.Process
类,以使分叉
在DGL核心库中与OpenMP一起工作。
API 的使用方式与原生模块完全相同,因此 DGL 不提供额外的文档。
此外,如果您的后端是PyTorch,此模块也将与
torch.multiprocessing
模块兼容。
|
在PyTorch分布式进程组的单个进程中调用函数,并与其他进程共享结果。 |
|
在共享内存中创建一个张量,该张量可以被同一 |