train/comms/pt/comms.py [384:392]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    self.collectiveArgs, self.collectiveArgs.dst_ranks[idx]
                )
            elif self.collectiveArgs.global_rank in self.collectiveArgs.dst_ranks:
                idx = self.collectiveArgs.dst_ranks.index(
                    self.collectiveArgs.global_rank
                )
                self.backendFuncs.recv(
                    self.collectiveArgs, self.collectiveArgs.src_ranks[idx]
                )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



train/comms/pt/comms.py [419:427]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    self.collectiveArgs, self.collectiveArgs.dst_ranks[idx]
                )
            elif self.collectiveArgs.global_rank in self.collectiveArgs.dst_ranks:
                idx = self.collectiveArgs.dst_ranks.index(
                    self.collectiveArgs.global_rank
                )
                self.backendFuncs.recv(
                    self.collectiveArgs, self.collectiveArgs.src_ranks[idx]
                )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -