Update lib/rocprofiler-sdk/hsa/queue.cpp (#788)

- fix heap-use-after-free [ROCm/rocprofiler-sdk commit: b7b2c294e0]
2024-04-18 00:05:55 -05:00
parent 3f9dc16864
commit 39550c1167
1 changed files with 6 additions and 0 deletions
@@ -200,6 +200,12 @@ WriteInterceptor(const void* packets,
                 void*                                 data,
                 hsa_amd_queue_intercept_packet_writer writer)
 {
+    if(registration::get_fini_status() > 0)
+    {
+        writer(packets, pkt_count);
+        return;
+    }
+
    using callback_record_t = Queue::queue_info_session_t::callback_record_t;

    // unique sequence id for the dispatch