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