Searched refs:colocations (Results 1 – 9 of 9) sorted by relevance
40 block->colocations.push_back(block); in MakeAllocationBlock()63 allocation_blocks[0]->colocations.push_back(allocation_blocks[1]); in TEST_F()64 allocation_blocks[1]->colocations.push_back(allocation_blocks[0]); in TEST_F()96 allocation_blocks[0]->colocations.push_back(allocation_blocks[1]); in TEST_F()97 allocation_blocks[1]->colocations.push_back(allocation_blocks[0]); in TEST_F()
43 allocation_block->colocations, [&](AllocationBlock* search) { in ImportAllocationBlocks()46 if (aliased_it != allocation_block->colocations.end()) { in ImportAllocationBlocks()47 buffer_intervals_[*aliased_it].colocations.push_back(allocation_block); in ImportAllocationBlocks()
45 std::vector<AllocationBlock*> colocations; member
87 absl::c_all_of(interval.colocations, IsValueAllowedInAlternateMemory); in IsIntervalAllowedInAlternateMemory()
150 interval.colocations = {++buffer.values().begin(), buffer.values().end()}; in FindCrossProgramPrefetchCandidate()1158 for (const HloValue* buffer_colocated : item->colocations) { in GetSortedColocatedIntervals()2139 for (const HloValue* colocation : prefetch_candidate->colocations) { in AllocateCrossProgramPrefetchBuffer()2241 colocation.colocations.push_back(inserted); in AllocateCrossProgramPrefetchBuffer()2243 inserted->colocations.push_back(&colocation); in AllocateCrossProgramPrefetchBuffer()2255 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*> colocations; in AllocateReservedScopedAllocations() local2269 interval.colocations = {}; in AllocateReservedScopedAllocations()2290 colocations.push_back(&repack_allocation_blocks_.back()); in AllocateReservedScopedAllocations()2300 colocations) { in AllocateReservedScopedAllocations()2301 repack_block->colocations = colocations; in AllocateReservedScopedAllocations()[all …]
385 absl::InlinedVector<const BufferType*, 2> colocations; member
569 buffer_intervals_[share_with].colocations.push_back(buffer); in ShareWith()585 for (const BufferType* buffer_colocated : item->colocations) { in GetTransitiveColocations()
5276 absl::flat_hash_set<int64_t> colocations; in Repack() local5278 for (const AllocationBlock* colocation : block->colocations) { in Repack()5280 colocations.insert(colocation->id); in Repack()5293 for (AllocationBlock* colocation : block->colocations) { in Repack()5531 EXPECT_TRUE(allocations.at(0)->colocations.size() == 1 || in TEST_P()5532 allocations.at(0)->colocations.size() == 3); in TEST_P()5533 EXPECT_EQ(allocations.at(1)->colocations.size(), 3); in TEST_P()5534 EXPECT_EQ(allocations.at(2)->colocations.size(), 3); in TEST_P()5535 EXPECT_TRUE(allocations.at(3)->colocations.size() == 1 || in TEST_P()5536 allocations.at(3)->colocations.size() == 3); in TEST_P()[all …]
1589 allocation_block.colocations = {}; in MakeRepackAllocationBlock()