From 52c50c165364a27fe1a8ed46a53dc7f726bc2f3a Mon Sep 17 00:00:00 2001
From: Wenkai Du <wenkai.du@amd.com>
Date: Thu, 10 Sep 2020 22:31:19 +0000
Subject: [PATCH] Add test for alltoallv API

[ROCm/rccl-tests commit: 3f1dfacc9503434560d6e16551865b7d8d816a96]
---
 projects/rccl-tests/src/alltoallv.cu | 4 ++++
 1 file changed, 4 insertions(+)

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<nranks; r++) {
@@ -145,6 +148,7 @@ testResult_t AlltoAllvRunColl(void* sendbuff, void* recvbuff, size_t count, nccl
     }
   }
   NCCLCHECK(ncclGroupEnd());
+#endif
   return testSuccess;
 #endif
 }