Home
last modified time | relevance | path

Searched refs:root_rank (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc207 int root_rank = -1; member
565 if (collective->status.ok() && collective->root_rank >= 0 && in AddParticipant()
567 collective->root_rank != context.source_rank) { in AddParticipant()
570 " already has root_rank ", collective->root_rank, in AddParticipant()
582 collective->root_rank = context.source_rank; in AddParticipant()
638 if (collective->root_rank == -1) { in RunCollective()
639 collective->root_rank = rank; in RunCollective()
640 } else if (collective->root_rank != rank) { in RunCollective()
642 "Inconsistent root rank ", collective->root_rank, " and GPU id ", in RunCollective()
647 << p->global_rank << " root_rank " << collective->root_rank; in RunCollective()
[all …]