From b251c8fdb2ed1f63401830f5eb3ce66b3bee41e9 Mon Sep 17 00:00:00 2001 From: sdashmiz Date: Thu, 5 Jan 2023 14:08:45 -0500 Subject: [PATCH] SWDEV-374368 - dependency list should not have duplicate Signed-off-by: sdashmiz Change-Id: I67e1c8203f10916b98408e2004e6c64e96b1933b --- hipamd/src/hip_graph.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp index 74c892497e..35b14e230e 100644 --- a/hipamd/src/hip_graph.cpp +++ b/hipamd/src/hip_graph.cpp @@ -36,11 +36,16 @@ inline hipError_t ihipGraphAddNode(hipGraphNode_t graphNode, hipGraph_t graph, const hipGraphNode_t* pDependencies, size_t numDependencies, bool capture = true) { graph->AddNode(graphNode); + std::unordered_set DuplicateDep; for (size_t i = 0; i < numDependencies; i++) { if ((!hipGraphNode::isNodeValid(pDependencies[i])) || (graph != pDependencies[i]->GetParentGraph())) { return hipErrorInvalidValue; } + if (DuplicateDep.find(pDependencies[i]) != DuplicateDep.end()) { + return hipErrorInvalidValue; + } + DuplicateDep.insert(pDependencies[i]); pDependencies[i]->AddEdge(graphNode); } if (capture == false) {