From 3a741787bf6998c4d711f2187bc0573392e9e8ce Mon Sep 17 00:00:00 2001 From: Nusrat Islam Date: Wed, 14 Jun 2023 13:36:15 -0500 Subject: [PATCH] device: use unroll factor based on platforms --- src/collectives/device/common.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/collectives/device/common.h b/src/collectives/device/common.h index a53ae58da5..6e563b811c 100644 --- a/src/collectives/device/common.h +++ b/src/collectives/device/common.h @@ -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()