diff --git a/src/collectives/device/common.h b/src/collectives/device/common.h index cedb3bb855..49eea4ee6e 100644 --- a/src/collectives/device/common.h +++ b/src/collectives/device/common.h @@ -11,7 +11,11 @@ #include "collectives.h" #include "devcomm.h" +#if defined(__gfx908__) +#define COLL_UNROLL 2 +#else #define COLL_UNROLL 4 +#endif #define NCCL_MAX_DEV_ARITY (NCCL_MAX_TREE_ARITY-1) // Using balanced tree instead of split tree diff --git a/src/collectives/device/sendrecv.h b/src/collectives/device/sendrecv.h index 5b6cf4b647..86c72f457a 100644 --- a/src/collectives/device/sendrecv.h +++ b/src/collectives/device/sendrecv.h @@ -204,6 +204,8 @@ struct RunWork { } else { #if defined(__gfx90a__) runRecv>(tid, nthreads, group, args); +#elif defined(__gfx908__) + runRecv>(tid, nthreads, group, args); #else runRecv>(tid, nthreads, group, args); #endif @@ -214,6 +216,8 @@ struct RunWork { } else { #if defined(__gfx90a__) runSend>(tid, nthreads, group, args); +#elif defined(__gfx908__) + runSend>(tid, nthreads, group, args); #else runSend>(tid, nthreads, group, args); #endif