From dcd0322f52fa0fbdf55f7b83412918194c644875 Mon Sep 17 00:00:00 2001
From: Satyanvesh Dittakavi <Satyanvesh.Dittakavi@amd.com>
Date: Wed, 12 Jan 2022 16:06:14 +0000
Subject: [PATCH] SWDEV-318128 - Add Null checks in
 hipGraphExecHostNodeSetParams API

Change-Id: I1363b16646b7629c827ace4db4c7b0da32486edb
---
 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 2dc35b1e2f..33c53229ea 100644
--- a/hipamd/src/hip_graph.cpp
+++ b/hipamd/src/hip_graph.cpp
@@ -1515,7 +1515,8 @@ hipError_t hipGraphHostNodeSetParams(hipGraphNode_t node, const hipHostNodeParam
 hipError_t hipGraphExecHostNodeSetParams(hipGraphExec_t hGraphExec, hipGraphNode_t node,
                                          const hipHostNodeParams* pNodeParams) {
   HIP_INIT_API(hipGraphExecHostNodeSetParams, hGraphExec, node, pNodeParams);
-  if (pNodeParams->fn == nullptr || pNodeParams->userData == nullptr) {
+  if (hGraphExec == nullptr || pNodeParams == nullptr ||
+      pNodeParams->fn == nullptr || pNodeParams->userData == nullptr) {
     HIP_RETURN(hipErrorInvalidValue);
   }
   hipGraphNode_t clonedNode = hGraphExec->GetClonedNode(node);