diff --git a/src/collectives/device/sendrecv.h b/src/collectives/device/sendrecv.h index 030b1af7c4..647378f2a3 100644 --- a/src/collectives/device/sendrecv.h +++ b/src/collectives/device/sendrecv.h @@ -202,13 +202,21 @@ struct RunWork { if (args->proto == NCCL_PROTO_LL) { runRecv(tid, nthreads, group, args); } else { +#if defined(__gfx90a__) + runRecv>(tid, nthreads, group, args); +#else runRecv>(tid, nthreads, group, args); +#endif } } else { if (args->proto == NCCL_PROTO_LL) { runSend(tid, nthreads, group, args); } else { +#if defined(__gfx90a__) + runSend>(tid, nthreads, group, args); +#else runSend>(tid, nthreads, group, args); +#endif } } }