mirrors / dmlc--dgl

项目文件夹

文件

T

Xin Yao 8d5d8962ad [Refactor] Replace third_party/nccl with PyTorch's NCCL backend (#4989 )

* expose GeneratePermutation

* add sparse_all_to_all_push

* add sparse_all_to_all_pull

* add unit test

* handle world_size=1

* remove python nccl wrapper

* remove the nccl dependency

* use pinned memory to speedup D2H copy

* fix lint

* resolve comments

* fix lint

* fix ut

* resolve comments

2023-03-08 12:59:10 +08:00

6 行

117 B

Python

原始文件 永久链接 Blame 文件历史

 """ CUDA wrappers """
 from .. import backend as F
 if F.get_preferred_backend() == "pytorch":
     from . import nccl