From 9925195afc967bc50f82aecaa32c3c09866fbe15 Mon Sep 17 00:00:00 2001
From: akolliasAMD <99202231+akolliasAMD@users.noreply.github.com>
Date: Thu, 21 Jul 2022 10:28:53 -0600
Subject: [PATCH] updated alltoallV test to not have any zero values (#12)

updated alltoallV test to not have any zero values between ranks
---
 src/alltoallv.cu | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/alltoallv.cu b/src/alltoallv.cu
index 7a39bcce7b..cb8fcaff0d 100644
--- a/src/alltoallv.cu
+++ b/src/alltoallv.cu
@@ -71,12 +71,12 @@ testResult_t AlltoAllvInitData(struct threadArgs* args, ncclDataType_t type, ncc
       size_t chunksize = data_count/nranks;
       for (int j=0; j<nranks; j++) {
 	size_t scount = 0, rcount = ((j+rank)%nranks)*chunksize;
-	if (j+rank == nranks-1)
+	if ((j+rank)%nranks == 0)
           rcount += (sendcount-chunksize*(nranks-1)*nranks/2);
 	size_t sdisp = 0;
 	for (int k=0; k<nranks; k++) {
 	  scount = ((k+j)%nranks)*chunksize;
-	  if (k+j == nranks-1)
+	  if ((k+j)%nranks == 0)
 	    scount += (sendcount-chunksize*(nranks-1)*nranks/2);
 	  if (k == rank)
 	    break;
@@ -119,7 +119,7 @@ testResult_t AlltoAllvRunColl(void* sendbuff, void* recvbuff, size_t count, nccl
   size_t chunksize = count*2/nranks;
   for (int i = 0; i < nranks; i++) {
       size_t scount = ((i+rank)%nranks)*chunksize;
-      if (i+rank == nranks-1)
+      if ((i+rank)%nranks == 0)
           scount += (count*nranks-chunksize*(nranks-1)*nranks/2);
       sendcounts[i+rank*MAX_ALLTOALLV_RANKS] = recvcounts[i+rank*MAX_ALLTOALLV_RANKS] = scount;
       sdispls[i+rank*MAX_ALLTOALLV_RANKS] = rdispls[i+rank*MAX_ALLTOALLV_RANKS] = disp;