Граф коммитов

6 Коммитов

Автор SHA1 Сообщение Дата
Edgar 0500f2f132 implementation of multi-rank support in rccl-tests. 2022-06-10 14:54:10 -04:00
Wenkai Du 213abee002 Merge remote-tracking branch 'nccl/master' into develop 2021-09-20 14:01:22 -07:00
David Addison f773748b46 Resync with NCCL 2.11
New operator: mulsum
New test: gather
2021-09-17 09:02:45 -07:00
Wenkai Du 346cb16442 Change scatter and gather bandwidth calculation to match alltoall (#7) 2020-07-06 17:12:50 -07:00
Wenkai Du 2813c96826 Add option to use alltoall, gather and scatter API
These APIs launche RCCL kernel implementation by default. If environmental
variable RCCL_ALLTOALL_KERNEL_DISABLE=1, then the APIs use wrapper around
ncclSend and ncclRecv.
2020-05-11 15:21:19 -07:00
Wenkai Du 6932a583e7 Add gather and scatter test 2020-04-10 14:45:10 -07:00