SWDEV-389648 - Find node and then erase.

Change-Id: Iecf78fe6b0645e58b4de6658b59ad1285946b698


[ROCm/clr commit: 3c5888922e]
Cette révision appartient à :
Jaydeep Patel
2023-04-10 06:31:13 +00:00
révisé par Jaydeepkumar Patel
Parent 3bab75e327
révision e404ad7bd8
+4 -1
Voir le fichier
@@ -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;