From e404ad7bd875c2cf3a14d6e7447c3590a04afe2b Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Mon, 10 Apr 2023 06:31:13 +0000
Subject: [PATCH] SWDEV-389648 - Find node and then erase.

Change-Id: Iecf78fe6b0645e58b4de6658b59ad1285946b698


[ROCm/clr commit: 3c5888922e7e990fe84c802ab6d077a23ef670d2]
---
 projects/clr/hipamd/src/hip_graph.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/projects/clr/hipamd/src/hip_graph.cpp b/projects/clr/hipamd/src/hip_graph.cpp
index 787ba9c69d..9be59fe76c 100644
--- a/projects/clr/hipamd/src/hip_graph.cpp
+++ b/projects/clr/hipamd/src/hip_graph.cpp
@@ -1052,7 +1052,10 @@ hipError_t hipStreamEndCapture_common(hipStream_t stream, hipGraph_t* pGraph) {
     std::vector<hipGraphNode_t> leafNodes = s->GetCaptureGraph()->GetLeafNodes();
     std::unordered_set<hipGraphNode_t> nodes = s->GetCaptureGraph()->GetManualNodesDuringCapture();
     for (auto node : nodes) {
-      leafNodes.erase(std::find(leafNodes.begin(), leafNodes.end(), node));
+      const auto& fnode = std::find(leafNodes.begin(), leafNodes.end(), node);
+      if (fnode != leafNodes.end()) {
+        leafNodes.erase(fnode);
+      }
     }
     const std::vector<hipGraphNode_t>& removedDepNodes = s->GetRemovedDependencies();
     bool foundInRemovedDep = false;