Searched refs:colocations (Results 1 – 9 of 9) sorted by relevance
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()
38 block->colocations.push_back(block); in MakeAllocationBlock()61 allocation_blocks[0]->colocations.push_back(allocation_blocks[1]); in TEST_F()62 allocation_blocks[1]->colocations.push_back(allocation_blocks[0]); in TEST_F()
45 std::vector<AllocationBlock*> colocations; member
112 absl::c_all_of(interval.colocations, IsValueAllowedInAlternateMemory); in IsIntervalAllowedInAlternateMemory()
107 interval.colocations = {++buffer.values().begin(), buffer.values().end()}; in FindCrossProgramPrefetchCandidate()875 for (const HloValue* buffer_colocated : item->colocations) { in GetSortedColocatedIntervals()1563 for (const HloValue* colocation : prefetch_candidate->colocations) { in AllocateCrossProgramPrefetchBuffer()1622 colocation.colocations.push_back(inserted); in AllocateCrossProgramPrefetchBuffer()1624 inserted->colocations.push_back(&colocation); in AllocateCrossProgramPrefetchBuffer()1902 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*> colocations; in FinalizeAllocations() local1911 colocations.push_back(&repack_allocation_blocks_.back()); in FinalizeAllocations()1914 colocations) { in FinalizeAllocations()1915 repack_block->colocations = colocations; in FinalizeAllocations()
379 std::vector<const BufferType*> colocations; member
583 buffer_intervals_[share_with].colocations.push_back(buffer); in ShareWith()599 for (const BufferType* buffer_colocated : item->colocations) { in GetTransitiveColocations()
4391 absl::flat_hash_set<int64> colocations; in Repack() local4393 for (const AllocationBlock* colocation : block->colocations) { in Repack()4395 colocations.insert(colocation->id); in Repack()4408 for (AllocationBlock* colocation : block->colocations) { in Repack()4646 EXPECT_TRUE(allocations.at(0)->colocations.size() == 1 || in TEST_P()4647 allocations.at(0)->colocations.size() == 3); in TEST_P()4648 EXPECT_EQ(allocations.at(1)->colocations.size(), 3); in TEST_P()4649 EXPECT_EQ(allocations.at(2)->colocations.size(), 3); in TEST_P()4650 EXPECT_TRUE(allocations.at(3)->colocations.size() == 1 || in TEST_P()4651 allocations.at(3)->colocations.size() == 3); in TEST_P()
1317 allocation_block.colocations = {}; in MakeRepackAllocationBlock()