Skip to content

Commit 9526df4

Browse files
mingzhe09088facebook-github-bot
authored andcommitted
fix gloo allreduce_bcube issue for platform010 switch
Summary: fixed a unit test failure in D37416148 after switch gloo to platform010. Reviewed By: meyering Differential Revision: D37423180 fbshipit-source-id: 1314386206483287a5e54933c96e786d051dc339
1 parent 5b14351 commit 9526df4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

gloo/allreduce_bcube.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,8 @@ class Node {
6868
for (int i = 0; i < steps; ++i) {
6969
peersPerStep_.emplace_back();
7070
}
71-
numElemsPerStep_.reserve(steps);
72-
ptrOffsetPerStep_.reserve(steps);
71+
numElemsPerStep_.resize(steps);
72+
ptrOffsetPerStep_.resize(steps);
7373
}
7474
/**
7575
* Get the rank of this node
@@ -295,7 +295,7 @@ class AllreduceBcube : public Algorithm {
295295
std::max(myRank_, destRank));
296296
sendDataBufs_[destRank] =
297297
pair->createSendBuffer(slot, ptrs_[0], bytes_);
298-
recvBufs_[bufIdx].reserve(recvSize);
298+
recvBufs_[bufIdx].resize(recvSize);
299299
recvDataBufs_[destRank] = pair->createRecvBuffer(
300300
slot, &recvBufs_[bufIdx][0], recvSize * sizeof(T));
301301
recvBufIdx_[destRank] = bufIdx;

0 commit comments

Comments
 (0)