Add test for alltoallv API

2020-09-10 22:31:19 +00:00
parent 0d1940e18e
commit 3f1dfacc95
1 changed files with 4 additions and 0 deletions
@@ -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<nranks; r++) {
@@ -145,6 +148,7 @@ testResult_t AlltoAllvRunColl(void* sendbuff, void* recvbuff, size_t count, nccl
    }
  }
  NCCLCHECK(ncclGroupEnd());
+#endif
  return testSuccess;
 #endif
 }