Fix typo in rank assignment (#59)

此提交包含在:
Wenkai Du
2024-02-15 12:04:38 -08:00
提交者 GitHub
父節點 28b05ac4b5
當前提交 7715a0cf1f
+1 -1
查看文件
@@ -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