diff --git a/projects/rccl-tests/src/alltoallv.cu b/projects/rccl-tests/src/alltoallv.cu index 7993059441..30577fea43 100644 --- a/projects/rccl-tests/src/alltoallv.cu +++ b/projects/rccl-tests/src/alltoallv.cu @@ -122,6 +122,9 @@ testResult_t AlltoAllvRunColl(void* sendbuff, void* recvbuff, size_t count, nccl #if NCCL_MAJOR < 2 || NCCL_MINOR < 7 printf("NCCL 2.7 or later is needed for alltoallv. This test was compiled with %d.%d.\n", NCCL_MAJOR, NCCL_MINOR); return testNcclError; +#else +#if defined(RCCL_ALLTOALLV) && defined(USE_RCCL_GATHER_SCATTER) + NCCLCHECK(ncclAllToAllv(sendbuff, sendcounts, sdispls, recvbuff, recvcounts, rdispls, type, comm, stream)); #else NCCLCHECK(ncclGroupStart()); for (int r=0; r