fix use after free

AMReX-Codes · Aug 27, 2024 · 020166b · 020166b
1 parent 34d5508
commit 020166b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Src/Base/AMReX_BArena.cpp b/Src/Base/AMReX_BArena.cpp
@@ -11,8 +11,8 @@ amrex::BArena::alloc (std::size_t sz_)
 void
 amrex::BArena::free (void* pt)
 {
-    std::free(pt);
     m_profiler.free(pt);
+    std::free(pt);
 }
 
 bool

diff --git a/Src/Base/AMReX_PArena.cpp b/Src/Base/AMReX_PArena.cpp
@@ -94,11 +94,11 @@ PArena::free (void* p)
 
 #if defined (AMREX_GPU_STREAM_ALLOC_SUPPORT)
     if (Gpu::Device::memoryPoolsSupported()) {
+        m_profiler.free(p);
         AMREX_HIP_OR_CUDA(
             AMREX_HIP_SAFE_CALL(hipFreeAsync(p, Gpu::gpuStream()));,
             AMREX_CUDA_SAFE_CALL(cudaFreeAsync(p, Gpu::gpuStream()));
         )
-        m_profiler.free(p);
     } else
 #endif
     {