D | heap_simulator_test.cc | 1026 std::pair<int64, int64> FindChunkCandidate(const HloValue* buffer, in FindChunkCandidate() function in xla::__anon44f19eea0111::GlobalDecreasingSizeBestFitHeapTest::InheritedGlobalDecreasingSizeBestFitHeap 1034 chunk_candidate_ = GlobalDecreasingSizeBestFitHeap::FindChunkCandidate( in FindChunkCandidate() 1316 EXPECT_EQ(pair(5, 10), heap_.FindChunkCandidate(buffer_a_, 5, 6, 10, 5)); in TEST_F() 1319 EXPECT_EQ(pair(0, 10), heap_.FindChunkCandidate(buffer_b_, 10, 3, 5)); in TEST_F() 1321 EXPECT_EQ(pair(10, 15), heap_.FindChunkCandidate(buffer_c_, 5, 2, 8)); in TEST_F() 1323 EXPECT_EQ(pair(0, 15), heap_.FindChunkCandidate(buffer_d_, 5, 0, 2, 10)); in TEST_F() 1326 EXPECT_EQ(pair(10, 20), heap_.FindChunkCandidate(buffer_e_, 10, 11, 13, 10)); in TEST_F() 1329 EXPECT_EQ(pair(20, 25), heap_.FindChunkCandidate(buffer_f_, 5, 3, 5, 20)); in TEST_F() 1332 EXPECT_EQ(pair(25, 35), heap_.FindChunkCandidate(buffer_g_, 10, 4, 8, 15)); in TEST_F()
|