Searched refs:local_rank (Results 1 – 2 of 2) sorted by relevance
125 for (int local_rank = 0; local_rank < num_ranks_per_node; ++local_rank) { in MakeReductionTestCase() local126 auto* device = GetDevice(num_ranks_per_node, node, local_rank); in MakeReductionTestCase()219 for (int local_rank = 0; local_rank < num_ranks_per_node; ++local_rank) { in MakeBroadcastTestCase() local220 auto* device = GetDevice(num_ranks_per_node, node, local_rank); in MakeBroadcastTestCase()221 if (node == src_node && local_rank == src_rank) { in MakeBroadcastTestCase()260 for (int local_rank = 0; local_rank < test_case->num_ranks_per_node; in VerifyResults() local261 ++local_rank) { in VerifyResults()263 GetDevice(test_case->num_ranks_per_node, node, local_rank); in VerifyResults()266 GlobalRank(test_case->num_ranks_per_node, node, local_rank); in VerifyResults()331 for (int local_rank = 0; local_rank < num_ranks_per_node; in RunMultiNodeAllReduceTest() local[all …]
281 int local_rank; member300 dr->local_rank = 0; // Will be populated later by OrderTaskDeviceMap. in BuildDevRecs()332 dr->local_rank = rank_it->second; in ParseRingOrder()370 dr->local_rank = next_rank; in OrderTaskDeviceMap()442 it.second.global_rank = it.second.local_rank + next_rank; in EstablishGlobalRank()