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

3 Коммитов

Автор SHA1 Сообщение Дата
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