Searched refs:gpu_ring_order (Results 1 – 6 of 6) sorted by relevance
51 const std::vector<int32>& gpu_ring_order, in RunCompleteDefaultRanking() argument61 if (!gpu_ring_order.empty()) { in RunCompleteDefaultRanking()62 ir.shared.instance.gpu_ring_order = ""; in RunCompleteDefaultRanking()63 for (int i = 0; i < static_cast<int32>(gpu_ring_order.size() - 1); in RunCompleteDefaultRanking()65 ir.shared.instance.gpu_ring_order = strings::StrCat( in RunCompleteDefaultRanking()66 ir.shared.instance.gpu_ring_order, gpu_ring_order[i], ","); in RunCompleteDefaultRanking()68 ir.shared.instance.gpu_ring_order = strings::StrCat( in RunCompleteDefaultRanking()69 ir.shared.instance.gpu_ring_order, gpu_ring_order.back()); in RunCompleteDefaultRanking()71 VLOG(2) << "gpu_ring_order " << ir.shared.instance.gpu_ring_order; in RunCompleteDefaultRanking()
99 const string* gpu_ring_order) in BaseCollectiveExecutor() argument104 gpu_ring_order_(gpu_ring_order) {} in BaseCollectiveExecutor()
205 void OrderTaskDeviceMap(const string& gpu_ring_order, TaskDeviceMap* tdm) { in OrderTaskDeviceMap() argument209 if (ParseRingOrder(gpu_ring_order, tdm)) return; in OrderTaskDeviceMap()298 OrderTaskDeviceMap(cp->instance.gpu_ring_order, &tdm); in EstablishGlobalRank()
279 cp->instance.gpu_ring_order = *gpu_ring_order_; in CompleteParamsAsync()
69 gpu_ring_order = other.gpu_ring_order; in operator =()
95 string gpu_ring_order = ""; member