From 5dab99ca30cc9d8ec06e17c175bbe67518910210 Mon Sep 17 00:00:00 2001
From: Sarbojit Sarkar <sarbojit.sarkar@amd.com>
Date: Wed, 2 Feb 2022 06:59:20 +0000
Subject: [PATCH] SWDEV-317787 - Added missing nullchecks

Change-Id: I4abe4f519c97122a96f668230398b10d4c4b9533
---
 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 ab8ec06a7e..697cc49a7a 100644
--- a/hipamd/src/hip_graph.cpp
+++ b/hipamd/src/hip_graph.cpp
@@ -1632,6 +1632,12 @@ hipError_t hipGraphExecUpdate(hipGraphExec_t hGraphExec, hipGraph_t hGraph,
                               hipGraphNode_t* hErrorNode_out,
                               hipGraphExecUpdateResult* updateResult_out) {
   HIP_INIT_API(hipGraphExecUpdate, hGraphExec, hGraph, hErrorNode_out, updateResult_out);
+  // parameter check
+  if (hGraphExec == nullptr || hGraph == nullptr || hErrorNode_out == nullptr ||
+    updateResult_out == nullptr) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
+
   std::vector<Node> newGraphNodes;
   hGraph->LevelOrder(newGraphNodes);
   std::vector<Node>& oldGraphExecNodes = hGraphExec->GetNodes();