Merge sendrecv kernel from NCCL 2.7.3

This commit was cherry-picked and modified from
https://github.com/NVIDIA/nccl/commit/5949d96f36d050e59d05872f8bbffd2549318e95
Este cometimento está contido em:
Wenkai Du cometeu 2020-06-29 08:47:46 -07:00
1 ascendente b90735c935
cometimento 964c4c2061
3 ficheiros modificados
+51 -31

Não existem ficheiros que correspondam à sua pesquisa

+1 -1
Ver ficheiro
@@ -82,7 +82,7 @@ DECL_ALL_COLLS
#define BROADCAST_CHUNKSTEPS 1
#define REDUCE_SLICESTEPS 1
#define REDUCE_CHUNKSTEPS 1
#define SENDRECV_SLICEFACTOR 4
#define SENDRECV_SLICEFACTOR 1
#define GATHER_SLICESTEPS 4
#define GATHER_CHUNKSTEPS 4
#define SCATTER_SLICESTEPS 4