database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

branch-2.1: [fix](memory) Fix UT ThreadMemTrackerMgrTest #44147 (#44936)

Browse Source 
Cherry-picked from #44147

Co-authored-by: Xinyi Zou <zouxinyi@selectdb.com>
This commit is contained in:
github-actions[bot]
2024-12-06 10:54:12 +08:00
committed by GitHub
parent 21e21f5e3b
commit 94746b764d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
be/test/runtime/memory/thread_mem_tracker_mgr_test.cpp
View File

@ -257,7 +257,7 @@ TEST_F(ThreadMemTrackerMgrTest, ReserveMemory) {
int64_t size1 = 4 * 1024;
int64_t size2 = 4 * 1024 * 1024;
int64_t size3 = size2 * 1024;
int64_t size3 = size2 * 2;
thread_context->attach_task(TUniqueId(), t, workload_group);
thread_context->consume_memory(size1);
@ -281,7 +281,7 @@ TEST_F(ThreadMemTrackerMgrTest, ReserveMemory) {
// std::abs(-size1 - size1) < SYNC_PROC_RESERVED_INTERVAL_BYTES, not update process_reserved_memory.
EXPECT_EQ(doris::GlobalMemoryArbitrator::process_reserved_memory(), size3 - size2);
thread_context->consume_memory(size2 * 1023);
thread_context->consume_memory(size2);
EXPECT_EQ(t->consumption(), size1 + size2 + size3);
EXPECT_EQ(doris::GlobalMemoryArbitrator::process_reserved_memory(), size1 + size1);
@ -408,7 +408,7 @@ TEST_F(ThreadMemTrackerMgrTest, NestedSwitchMemTrackerReserveMemory) {
int64_t size1 = 4 * 1024;
int64_t size2 = 4 * 1024 * 1024;
int64_t size3 = size2 * 1024;
int64_t size3 = size2 * 2;
thread_context->attach_task(TUniqueId(), t1, workload_group);
auto st = thread_context->try_reserve_memory(size3);