6 Commits

Author	SHA1	Message	Date
Jiaming Yuan	3f64b4fde3	[coll] Add global functions. (#10203)	2024-04-19 03:17:23 +08:00
Jiaming Yuan	8bad677c2f	Update collective implementation. (#10152) ... * Update collective implementation. - Cleanup resource during `Finalize` to avoid handling threads in destructor. - Calculate the size for allgather automatically. - Use simple allgather for small (smaller than the number of worker) allreduce.	2024-03-30 18:57:31 +08:00
Jiaming Yuan	6fd4a30667	[coll] Increase timeout for allgather test. (#9777)	2023-11-09 05:26:40 +08:00
Jiaming Yuan	6755179e77	[coll] Add nccl. (#9726)	2023-10-28 16:33:58 +08:00
Jiaming Yuan	5d1bcde719	[coll] allgatherv. (#9688)	2023-10-19 03:13:50 +08:00
Jiaming Yuan	4c0e4422d0	[coll] allgather. (#9681)	2023-10-18 10:22:18 +08:00