Searched refs:subdiv_permutations (Results 1 – 10 of 10) sorted by relevance
119 col_params->instance.impl_details.subdiv_permutations.resize(num_subdivs); in InitializeCollectiveParams()129 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()161 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()248 for (int i = 0; i < impl.subdiv_permutations[subdiv].size(); i++) { in TreeSendTo()249 if (impl.subdiv_permutations[subdiv][i] >= 0) { in TreeSendTo()304 for (int r : col_params_->instance.impl_details.subdiv_permutations[si]) { in RunTree()416 col_params_->instance.impl_details.subdiv_permutations[subdiv][dst_rank]; in DispatchSend()436 col_params_->instance.impl_details.subdiv_permutations[subdiv][src_rank]; in DispatchRecv()
65 cp->instance.impl_details.subdiv_permutations.push_back( \282 col_params_->instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()288 col_params_->instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()293 .subdiv_permutations[subdiv_i][rank] = i; in Init()300 col_params_->instance.impl_details.subdiv_permutations[subdiv_i]) in Init()308 col_params_->instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()313 << col_params_->instance.impl_details.subdiv_permutations.size(); in Init()317 col_params_->instance.impl_details.subdiv_permutations[subdiv_i][rank] = in Init()323 col_params_->instance.impl_details.subdiv_permutations[subdiv_i]) in Init()444 cp.instance.impl_details.subdiv_permutations in RunTest()[all …]
201 << &col_params->instance.impl_details.subdiv_permutations; in InitializeCollectiveParams()202 col_params->instance.impl_details.subdiv_permutations.resize( in InitializeCollectiveParams()209 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()331 .subdiv_permutations[subdiv_idx][recv_from_rank]; in InitRingField()333 .subdiv_permutations[subdiv_idx][send_to_rank]; in InitRingField()391 .subdiv_permutations[rf->subdiv_idx][send_to_rank]; in DispatchSend()
69 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()79 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()83 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
197 col_params_->instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()240 col_params_->instance.impl_details.subdiv_permutations[sdi].push_back( in Init()375 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()382 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()410 col_params_->instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()415 .subdiv_permutations[sdi] in DeviceInstance()419 col_params_->instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
65 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()75 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()79 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
220 col_params_->instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()264 col_params_->instance.impl_details.subdiv_permutations[sdi].push_back( in Init()400 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()407 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()436 col_params_->instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()441 .subdiv_permutations[sdi] in DeviceInstance()445 col_params_->instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
56 col_params.instance.impl_details.subdiv_permutations; in SubdivPermDebugString()
86 impl_details.subdiv_permutations.clear(); in operator =()87 for (auto p : other.impl_details.subdiv_permutations) { in operator =()88 impl_details.subdiv_permutations.push_back( in operator =()113 for (const auto& p : impl_details.subdiv_permutations) { in ToString()
93 std::vector<std::vector<int>> subdiv_permutations; member