From bc2a4f84502188fce9d1894ab5a09d3c3992fabe Mon Sep 17 00:00:00 2001 From: Ioannis Assiouras Date: Thu, 2 Nov 2023 11:32:34 +0000 Subject: [PATCH] SWDEV-430116 - Fix segfault in Heap::EraseAllocaton Change-Id: I8c49af081079307118562485cb0d5b3e8a9d906a --- hipamd/src/hip_mempool_impl.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hipamd/src/hip_mempool_impl.cpp b/hipamd/src/hip_mempool_impl.cpp index afdc35e3eb..230c60ea50 100644 --- a/hipamd/src/hip_mempool_impl.cpp +++ b/hipamd/src/hip_mempool_impl.cpp @@ -86,8 +86,8 @@ bool Heap::RemoveMemory(amd::Memory* memory, MemoryTimestamp* ts) { std::unordered_map::iterator Heap::EraseAllocaton(std::unordered_map::iterator& it) { const device::Memory* dev_mem = it->first->getDeviceMemory(*device_->devices()[0]); - amd::SvmBuffer::free(it->first->getContext(), reinterpret_cast(dev_mem->virtualAddress())); total_size_ -= it->first->getSize(); + amd::SvmBuffer::free(it->first->getContext(), reinterpret_cast(dev_mem->virtualAddress())); // Clear HIP event it->second.SetEvent(nullptr); // Remove the allocation from the map