Searched refs:subdiv_rank (Results 1 – 9 of 9) sorted by relevance
119 col_params->subdiv_rank.reserve(num_subdivs); in InitializeCollectiveParams()146 if (participate) col_params->subdiv_rank.push_back(ti); in InitializeCollectiveParams()149 if (col_params->subdiv_rank.empty()) col_params->subdiv_rank.push_back(-1); in InitializeCollectiveParams()169 col_params->subdiv_rank.push_back(di); in InitializeCollectiveParams()174 if (!participate) col_params->subdiv_rank.push_back(-1); in InitializeCollectiveParams()219 DCHECK_LT(subdiv, static_cast<int>(cp.subdiv_rank.size())); in TreeRecvFrom()220 int my_rank = cp.subdiv_rank[subdiv]; in TreeRecvFrom()239 DCHECK_LT(subdiv, static_cast<int>(cp.subdiv_rank.size())); in TreeSendTo()240 int my_rank = cp.subdiv_rank[subdiv]; in TreeSendTo()291 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in RunTree()[all …]
64 cp.subdiv_rank = {R}; \269 col_params_.subdiv_rank.resize(num_subdivs); in Init()522 cp->subdiv_rank.clear(); in RunSubdivPermsTest()529 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()575 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank; in DeviceInstance()585 col_params_.subdiv_rank.resize(num_subdivs); in DeviceInstance()594 col_params_.subdiv_rank[si] = perm_rank; in DeviceInstance()597 for (int r : col_params_.subdiv_rank) { in DeviceInstance()603 col_params_.subdiv_rank[0] == impl.subdiv_source_rank[0]; in DeviceInstance()
187 col_params_.subdiv_rank.resize(num_subdivs); in Init()192 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride; in Init()367 cp->subdiv_rank.clear(); in RunSubdivPermsTest()373 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()393 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank; in DeviceInstance()395 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size()); in DeviceInstance()411 col_params_.subdiv_rank[sdi] = r; in DeviceInstance()
209 col_params_.subdiv_rank.resize(num_subdivs); in Init()214 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride; in Init()390 cp->subdiv_rank.clear(); in RunSubdivPermsTest()396 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()417 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank; in DeviceInstance()419 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size()); in DeviceInstance()435 col_params_.subdiv_rank[sdi] = r; in DeviceInstance()
203 col_params->subdiv_rank.resize( in InitializeCollectiveParams()231 col_params->subdiv_rank[sdi] = rank; in InitializeCollectiveParams()322 rf->rank = col_params_->subdiv_rank[subdiv_idx]; in InitRingField()
71 for (auto d : col_params.subdiv_rank) strings::StrAppend(&buf, d, " "); in SubdivPermDebugString()
101 Tensor alias_chunk(ca_->ChunkAlias(col_params_->subdiv_rank[0])); in Run()
144 for (const auto& r : subdiv_rank) { in ToString()
121 std::vector<int> subdiv_rank; member