4 Commits

Author SHA1 Message Date
Andrei Ivanov
64df37f7eb Improving gpu_cache test. (#6195) 2023-08-23 13:46:08 +08:00
Muhammed Fatih BALIN
69a532c1ab [Feature] Gpu cache for node and edge data (#4341)
Co-authored-by: xiny <xiny@nvidia.com>
2023-07-24 13:17:10 +08:00
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
Hongzhi (Steve), Chen
d9e550369f [Misc] Reorg python test according to the guideline. (#5307)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-17 16:08:20 +08:00