From afae35b0fd1b69be7e2a9aa689fb626b77b182e3 Mon Sep 17 00:00:00 2001 From: Yifan Zhang Date: Wed, 15 Feb 2023 18:20:22 +0800 Subject: [PATCH] Avoid memory leak when rocrtstFunc.Memory_Available fails Assert abort the test thread w/ memPtr1 allocated. Free memPtr1 to avoid memory leak. Signed-off-by: Yifan Zhang Change-Id: I4e1a202c1acb9ba71a23e112254f875bf5a0abcf --- rocrtst/suites/functional/memory_basic.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/rocrtst/suites/functional/memory_basic.cc b/rocrtst/suites/functional/memory_basic.cc index 715eb60328..0f2dc99dd6 100755 --- a/rocrtst/suites/functional/memory_basic.cc +++ b/rocrtst/suites/functional/memory_basic.cc @@ -329,6 +329,7 @@ void MemoryTest::MemAvailableTest(hsa_agent_t ag, hsa_amd_memory_pool_t pool) { uint64_t allocate_sz2 = (0.8 * ag_avail_memory_after * gran_sz) / gran_sz; err = hsa_amd_memory_pool_allocate(pool, allocate_sz2, 0, &memPtr2); + if (err != HSA_STATUS_SUCCESS) hsa_memory_free(memPtr1); ASSERT_EQ(err, HSA_STATUS_SUCCESS); err = hsa_agent_get_info(ag, (hsa_agent_info_t)HSA_AMD_AGENT_INFO_MEMORY_AVAIL,