Merge pull request #775 from nusislam/unroll-1

device: use unroll factor based on platforms
2023-06-14 17:42:38 -05:00
parent f35777e9b0 3a741787bf
commit fd6553bcf8
1 changed files with 5 additions and 0 deletions
@@ -11,7 +11,12 @@
 #include "collectives.h"
 #include "devcomm.h"

+#if defined(__gfx940__)
 #define COLL_UNROLL 4
+#else
+#define COLL_UNROLL 2
+#endif
+
 #define NCCL_MAX_DEV_ARITY (NCCL_MAX_TREE_ARITY-1)  // Using balanced tree instead of split tree

 #define __syncwarp()