Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Saeed Maleki committed Nov 18, 2023
1 parent 7a283f7 commit 6e28973
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions python/benchmark/allreduce.cu
Original file line number Diff line number Diff line change
Expand Up @@ -177,11 +177,11 @@ __device__ void allreduce1_helper(mscclpp::SmChannelDeviceHandle* smChans, TYPE*
tmp += val;
}
if (READ_ONLY == 0) {
for (int index = 0; index < nPeer; ++index) {
int peerIdx = (index + rank);
if (peerIdx >= nPeer) peerIdx -= nPeer;
smChans[peerIdx].write<TYPE>(idx, tmp);
}
for (int index = 0; index < nPeer; ++index) {
int peerIdx = (index + rank);
if (peerIdx >= nPeer) peerIdx -= nPeer;
smChans[peerIdx].write<TYPE>(idx, tmp);
}
}
buff[idx] = tmp;
}
Expand Down

0 comments on commit 6e28973

Please sign in to comment.