Merge sendrecv kernel from NCCL 2.7.3

This commit was cherry-picked and modified from
https://github.com/NVIDIA/nccl/commit/5949d96f36d050e59d05872f8bbffd2549318e95
此提交包含在:
Wenkai Du committed 2020-06-29 08:47:46 -07:00
1 父節點 b90735c935
當前提交 964c4c2061
3 file changed
+51 -31

No files matched your search

+1 -1
查看文件
@@ -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