master

GGitHub[Refactor] Replace third_party/nccl with PyTorch's NCCL backend (#4989)

8d5d8962创建于 2023年3月8日历史提交

文件	最后提交记录	最后更新时间
test_nccl.py	[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	3 年前