Skip to content

Commit d02e227

Browse files
committed
Fix
1 parent b25dd61 commit d02e227

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

mooncake-ep/src/mooncake_backend.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -476,14 +476,14 @@ void MooncakeBackend::extendGroupSizeTo(int size) {
476476
}
477477

478478
std::vector<bool> MooncakeBackend::getPeerState(const std::vector<int>& ranks) {
479-
bool peerConnectedBackup[kMaxNumRanks];
479+
bool activeRanksBackup[kMaxNumRanks];
480480
while (true) {
481481
std::vector<int> input;
482482
for (const int rank : ranks) {
483483
input.push_back(meta_.peerConnected[rank]);
484484
}
485485
for (int i = 0; i < meta_.size; i++) {
486-
peerConnectedBackup[i] = meta_.peerConnected[i];
486+
activeRanksBackup[i] = meta_.activeRanks[i];
487487
}
488488

489489
std::vector<at::Tensor> tensors;
@@ -500,15 +500,15 @@ std::vector<bool> MooncakeBackend::getPeerState(const std::vector<int>& ranks) {
500500
at::cuda::getCurrentCUDAStream(tensors[0].device().index());
501501
cudaStreamSynchronize(stream);
502502
}
503-
bool peerConnectedChanged = false;
503+
bool activeRanksChanged = false;
504504
for (int i = 0; i < meta_.size; i++) {
505-
if (peerConnectedBackup[i] != meta_.peerConnected[i]) {
506-
peerConnectedChanged = true;
505+
if (activeRanksBackup[i] != meta_.activeRanks[i]) {
506+
activeRanksChanged = true;
507507
break;
508508
}
509509
}
510510

511-
if (!peerConnectedChanged) {
511+
if (!activeRanksChanged) {
512512
std::vector<bool> output;
513513
for (int i = 0; i < tensors[0].size(0); ++i) {
514514
output.push_back(tensors[0].cpu()[i].item<int>() != 0);

0 commit comments

Comments
 (0)