Searched refs:subdiv_permutations (Results 1 – 10 of 10) sorted by relevance
118 col_params->instance.impl_details.subdiv_permutations.resize(num_subdivs); in InitializeCollectiveParams()128 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()405 col_params_->instance.impl_details.subdiv_permutations[subdiv][dst_rank]; in DispatchSend()424 col_params_->instance.impl_details.subdiv_permutations[subdiv][src_rank]; in DispatchRecv()
62 cp.instance.impl_details.subdiv_permutations.push_back( \268 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()274 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()279 .subdiv_permutations[subdiv_i][rank] = i; in Init()286 col_params_.instance.impl_details.subdiv_permutations[subdiv_i]) in Init()294 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()299 << col_params_.instance.impl_details.subdiv_permutations.size(); in Init()303 col_params_.instance.impl_details.subdiv_permutations[subdiv_i][rank] = in Init()309 col_params_.instance.impl_details.subdiv_permutations[subdiv_i]) in Init()430 cp.instance.impl_details.subdiv_permutations in RunTest()[all …]
200 << &col_params->instance.impl_details.subdiv_permutations; in InitializeCollectiveParams()201 col_params->instance.impl_details.subdiv_permutations.resize( in InitializeCollectiveParams()208 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()329 .subdiv_permutations[subdiv_idx][recv_from_rank]; in InitRingField()331 .subdiv_permutations[subdiv_idx][send_to_rank]; in InitRingField()389 .subdiv_permutations[rf->subdiv_idx][send_to_rank]; in DispatchSend()
67 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()77 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()81 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
186 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()229 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back( in Init()366 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()372 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()401 col_params_.instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()406 .subdiv_permutations[sdi] in DeviceInstance()410 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
59 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()69 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()73 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
208 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()251 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back( in Init()389 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()395 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()425 col_params_.instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()430 .subdiv_permutations[sdi] in DeviceInstance()434 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
56 col_params.instance.impl_details.subdiv_permutations; in SubdivPermDebugString()
74 impl_details.subdiv_permutations.clear(); in operator =()75 for (auto p : other.impl_details.subdiv_permutations) { in operator =()76 impl_details.subdiv_permutations.push_back( in operator =()110 for (const auto& p : impl_details.subdiv_permutations) { in ToString()
71 std::vector<std::vector<int>> subdiv_permutations; member