Searched refs:subdiv_source_rank (Results 1 – 6 of 6) sorted by relevance
121 col_params->instance.impl_details.subdiv_source_rank.reserve(num_subdivs); in InitializeCollectiveParams()150 col_params->instance.impl_details.subdiv_source_rank.push_back(source_task); in InitializeCollectiveParams()175 col_params->instance.impl_details.subdiv_source_rank.push_back( in InitializeCollectiveParams()180 CHECK_GE(col_params->instance.impl_details.subdiv_source_rank[sri], 0); in InitializeCollectiveParams()224 DCHECK_LT(subdiv, static_cast<int>(impl.subdiv_source_rank.size())); in TreeRecvFrom()225 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeRecvFrom()244 DCHECK_LT(subdiv, static_cast<int>(impl.subdiv_source_rank.size())); in TreeSendTo()245 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeSendTo()301 int source_rank = col_params_->instance.impl_details.subdiv_source_rank[si]; in RunTree()
64 cp->instance.impl_details.subdiv_source_rank = {S}; \445 [0][cp.instance.impl_details.subdiv_source_rank[0]]; in RunTest()528 cp->instance.impl_details.subdiv_source_rank.clear(); in RunSubdivPermsTest()537 cp->instance.impl_details.subdiv_source_rank); in RunSubdivPermsTest()586 impl.subdiv_source_rank.resize(num_subdivs, 0); in DeviceInstance()605 col_params_->subdiv_rank[0] == impl.subdiv_source_rank[0]; in DeviceInstance()
74 for (auto src : col_params.instance.impl_details.subdiv_source_rank) in SubdivPermDebugString()
203 EXPECT_EQ(cps[i]->instance.impl_details.subdiv_source_rank.size(), 0); in TEST_F()
91 impl_details.subdiv_source_rank.assign( in operator =()92 other.impl_details.subdiv_source_rank.begin(), in operator =()93 other.impl_details.subdiv_source_rank.end()); in operator =()120 if (!impl_details.subdiv_source_rank.empty()) { in ToString()122 for (const auto& r : impl_details.subdiv_source_rank) { in ToString()
95 std::vector<int> subdiv_source_rank; // rank of source in each subdiv member