Merge pull request #757 from nusislam/unroll

device: change unroll factor [ROCm/rccl commit: 7519ecb476]
2023-05-30 14:19:32 -05:00
parent fd64cb242d 325ff1dc11
commit f4d63d21ca
1 changed files with 1 additions and 1 deletions
@@ -11,7 +11,7 @@
 #include "collectives.h"
 #include "devcomm.h"

-#define COLL_UNROLL 2
+#define COLL_UNROLL 4
 #define NCCL_MAX_DEV_ARITY (NCCL_MAX_TREE_ARITY-1)  // Using balanced tree instead of split tree

 #define __syncwarp()