SWDEV-389648 - Find node and then erase.
Change-Id: Iecf78fe6b0645e58b4de6658b59ad1285946b698
[ROCm/clr commit: 3c5888922e]
Cette révision appartient à :
révisé par
Jaydeepkumar Patel
Parent
3bab75e327
révision
e404ad7bd8
@@ -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;
|
||||
|
||||
Référencer dans un nouveau ticket
Bloquer un utilisateur