Merge sendrecv kernel from NCCL 2.7.3
This commit was cherry-picked and modified from https://github.com/NVIDIA/nccl/commit/5949d96f36d050e59d05872f8bbffd2549318e95
Este commit está contenido en:
1 padre
b90735c935
commit
964c4c2061
3 files changed
+51
-31
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user