@@ -476,14 +476,14 @@ void MooncakeBackend::extendGroupSizeTo(int size) {
476476}
477477
478478std::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