From 7715a0cf1ff638b81ca75f9166b0e2eff135bb37 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Thu, 15 Feb 2024 12:04:38 -0800
Subject: [PATCH] Fix typo in rank assignment (#59)

---
 src/common.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/common.cu b/src/common.cu
index e49c0d3f52..d71f0a7a35 100644
--- a/src/common.cu
+++ b/src/common.cu
@@ -1152,7 +1152,7 @@ testResult_t run() {
        for (int ii=0; ii<nGpus*nThreads; ii++) {
          HIPCHECK(hipSetDevice(gpus[ii]));
 	 if (!enable_multiranks) {
-	   NCCLCHECK(ncclCommInitRank(comms+ii, ncclProcs*nThreads*nGpus, ncclId, proc*nThreads*nGpus+ii));
+	   NCCLCHECK(ncclCommInitRank(comms+ii, ncclProcs*nThreads*nGpus, ncclId, ncclProc*nThreads*nGpus+ii));
 	 }
 #ifdef RCCL_MULTIRANKPERGPU
 	 else