From d27c33aecc554132a40a8df5a89b6b2bb267371c Mon Sep 17 00:00:00 2001
From: Payam <payam.ghafari@amd.com>
Date: Tue, 15 Feb 2022 09:49:35 -0500
Subject: [PATCH] SWDEV-313492 - adding Graph null checks

Change-Id: I00ae51ea8318567b4f6aab1568096d25cc0d373b
---
 hipamd/src/hip_graph.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp
index 8a9484926d..837eb39b67 100644
--- a/hipamd/src/hip_graph.cpp
+++ b/hipamd/src/hip_graph.cpp
@@ -887,8 +887,14 @@ hipError_t hipGraphAddChildGraphNode(hipGraphNode_t* pGraphNode, hipGraph_t grap
 }
 
 hipError_t ihipGraphInstantiate(hipGraphExec_t* pGraphExec, hipGraph_t graph) {
+  if (pGraphExec == nullptr || graph == nullptr) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
   std::unordered_map<Node, Node> clonedNodes;
   hipGraph_t clonedGraph = graph->clone(clonedNodes);
+  if (clonedGraph == nullptr) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
   std::vector<std::vector<Node>> parallelLists;
   std::unordered_map<Node, std::vector<Node>> nodeWaitLists;
   clonedGraph->GetRunList(parallelLists, nodeWaitLists);