From 94e219cd02091908cdb70ba12dca501c656ec010 Mon Sep 17 00:00:00 2001
From: darren-amd <darren.lao@amd.com>
Date: Thu, 29 May 2025 13:39:21 -0400
Subject: [PATCH 1/2] Add shutting down flag on thread

---
 source/timemory/process/threading.hpp | 11 +++++++++++
 source/timemory/sampling/sampler.cpp  |  4 ++--
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/source/timemory/process/threading.hpp b/source/timemory/process/threading.hpp
index ba15ad13c..7a2d61cba 100644
--- a/source/timemory/process/threading.hpp
+++ b/source/timemory/process/threading.hpp
@@ -264,6 +264,17 @@ struct affinity
 //
 //--------------------------------------------------------------------------------------//
 //
+inline thread_local bool is_shutting_down = false;
+
+inline void set_is_shutting_down(bool shutting_down = true)
+{
+    is_shutting_down = is_shutting_down;
+}
+
+inline bool get_is_shutting_down()
+{
+    return is_shutting_down;
+}
 }  // namespace threading
 }  // namespace tim
 
diff --git a/source/timemory/sampling/sampler.cpp b/source/timemory/sampling/sampler.cpp
index 6d99781ab..10c4049a7 100644
--- a/source/timemory/sampling/sampler.cpp
+++ b/source/timemory/sampling/sampler.cpp
@@ -380,7 +380,7 @@ template <template <typename...> class CompT, size_t N, typename... Types>
 void
 sampler<CompT<Types...>, N>::execute(int signum)
 {
-    if(!trait::runtime_enabled<this_type>::get())
+    if(!trait::runtime_enabled<this_type>::get() || threading::get_is_shutting_down())
         return;
 
     // save errno
@@ -411,7 +411,7 @@ template <template <typename...> class CompT, size_t N, typename... Types>
 void
 sampler<CompT<Types...>, N>::execute(int signum, siginfo_t* _info, void* _data)
 {
-    if(!trait::runtime_enabled<this_type>::get())
+    if(!trait::runtime_enabled<this_type>::get() || threading::get_is_shutting_down())
         return;
 
     // save errno

From 61f13795b9f2211778c4eaed15f91f99ccaf2cc1 Mon Sep 17 00:00:00 2001
From: darren-amd <darren.lao@amd.com>
Date: Thu, 29 May 2025 14:15:56 -0400
Subject: [PATCH 2/2] Fix shutting down setter

---
 source/timemory/process/threading.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/source/timemory/process/threading.hpp b/source/timemory/process/threading.hpp
index 7a2d61cba..b858ce977 100644
--- a/source/timemory/process/threading.hpp
+++ b/source/timemory/process/threading.hpp
@@ -268,7 +268,7 @@ inline thread_local bool is_shutting_down = false;
 
 inline void set_is_shutting_down(bool shutting_down = true)
 {
-    is_shutting_down = is_shutting_down;
+    is_shutting_down = shutting_down;
 }
 
 inline bool get_is_shutting_down()