Searched refs:global_rank (Results 1 – 4 of 4) sorted by relevance
70 Tensor* output, int global_rank, DoneCallback done_callback) in Participant()80 global_rank(global_rank), in Participant()120 const int global_rank; member
265 const int global_rank = in VerifyResults() local267 const Tensor& out_gpu = test_case->outs[global_rank]; in VerifyResults()273 VLOG(1) << "Verifying rank " << global_rank << " expected shape " in VerifyResults()336 const int global_rank = in RunMultiNodeAllReduceTest() local339 device->executor(), stream, info, &test_case->ins[global_rank], in RunMultiNodeAllReduceTest()340 &test_case->outs[global_rank], global_rank, in RunMultiNodeAllReduceTest()348 << global_rank; in RunMultiNodeAllReduceTest()385 const int global_rank = in RunMultiNodeBroadcastTest() local387 auto* input = global_rank == src_global_rank in RunMultiNodeBroadcastTest()388 ? &test_case->ins[global_rank] in RunMultiNodeBroadcastTest()[all …]
280 return a->global_rank < b->global_rank; in GetCommunicator()405 const int rank = collective->participants[i]->global_rank >= 0 in GetCommunicator()406 ? collective->participants[i]->global_rank in GetCommunicator()629 const int rank = p->global_rank >= 0 ? p->global_rank : i; in RunCollective()647 << p->global_rank << " root_rank " << collective->root_rank; in RunCollective()
282 int global_rank; member301 dr->global_rank = 0; // Will be populated later by EstablishGlobalRank. in BuildDevRecs()442 it.second.global_rank = it.second.local_rank + next_rank; in EstablishGlobalRank()584 new_device_names[dr.global_rank] = gp->device_names[dr.original_rank]; in CompleteDefaultRanking()585 new_task_names[dr.global_rank] = gp->task_names[dr.original_rank]; in CompleteDefaultRanking()