From a427d3cd10d97b15213de5bd025d202806e13d61 Mon Sep 17 00:00:00 2001
From: Rakesh Roy <rakesh.roy@amd.com>
Date: Thu, 12 Jan 2023 18:39:34 +0530
Subject: [PATCH] SWDEV-374369 - Fix hipGraphAdd*Node* APIs invalid behavior

- Return hipErrorInvalidValue if dependencies array contains a node from
  different graph

Change-Id: I0dcfaaaeb50a89848ac546a0b86b185977e2128c
---
 hipamd/src/hip_graph.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp
index 43e833bdf2..c74ce1b33f 100644
--- a/hipamd/src/hip_graph.cpp
+++ b/hipamd/src/hip_graph.cpp
@@ -32,7 +32,8 @@ inline hipError_t ihipGraphAddNode(hipGraphNode_t graphNode, hipGraph_t graph,
                                    const hipGraphNode_t* pDependencies, size_t numDependencies) {
   graph->AddNode(graphNode);
   for (size_t i = 0; i < numDependencies; i++) {
-    if (!hipGraphNode::isNodeValid(pDependencies[i])) {
+    if ((!hipGraphNode::isNodeValid(pDependencies[i])) ||
+        (graph != pDependencies[i]->GetParentGraph())) {
       return hipErrorInvalidValue;
     }
     pDependencies[i]->AddEdge(graphNode);