From ec391d72491e68afab2408976bf9ea4bcb63920b Mon Sep 17 00:00:00 2001 From: Anusha GodavarthySurya Date: Mon, 11 Mar 2024 13:19:11 +0000 Subject: [PATCH] SWDEV-436126 - hipStreamDestroy to remove stream from capture list Change-Id: Ie735f27b9c5df4faf39bd4be10fbe4401fd88cdb [ROCm/clr commit: 476dc4c87c286583b836e5bd049422965696856f] --- projects/clr/hipamd/src/hip_stream.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/projects/clr/hipamd/src/hip_stream.cpp b/projects/clr/hipamd/src/hip_stream.cpp index 9315082883..9d0475bc93 100644 --- a/projects/clr/hipamd/src/hip_stream.cpp +++ b/projects/clr/hipamd/src/hip_stream.cpp @@ -496,6 +496,13 @@ hipError_t hipStreamDestroy(hipStream_t stream) { g_captureStreams.erase(g_it); } } + { + amd::ScopedLock lock(g_streamSetLock); + const auto& g_it = std::find(g_allCapturingStreams.begin(), g_allCapturingStreams.end(), s); + if (g_it != g_allCapturingStreams.end()) { + g_allCapturingStreams.erase(g_it); + } + } const auto& l_it = std::find(hip::tls.capture_streams_.begin(), hip::tls.capture_streams_.end(), s); if (l_it != hip::tls.capture_streams_.end()) {