From 6eca010e9e017ddda78e6d4dbc2c940331df0354 Mon Sep 17 00:00:00 2001 From: sdashmiz Date: Tue, 15 Mar 2022 12:44:09 -0400 Subject: [PATCH] SWDEV-315402: Null param check for pGraphNode Signed-off-by: sdashmiz Change-Id: I4cd3a539cff1a01b4786d4edbc4201cd68346e0e [ROCm/clr commit: 8e81b8495cb4f2bc38c80d7c6202b9161e685536] --- projects/clr/hipamd/src/hip_graph.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/projects/clr/hipamd/src/hip_graph.cpp b/projects/clr/hipamd/src/hip_graph.cpp index 97f3a33bd1..03ad8d5ee7 100644 --- a/projects/clr/hipamd/src/hip_graph.cpp +++ b/projects/clr/hipamd/src/hip_graph.cpp @@ -1603,7 +1603,8 @@ hipError_t hipGraphAddEventRecordNode(hipGraphNode_t* pGraphNode, hipGraph_t gra hipEvent_t event) { HIP_INIT_API(hipGraphAddEventRecordNode, pGraphNode, graph, pDependencies, numDependencies, event); - if (graph == nullptr || (numDependencies > 0 && pDependencies == nullptr) || event == nullptr) { + if (pGraphNode == nullptr || graph == nullptr || (numDependencies > 0 && pDependencies == nullptr) + || event == nullptr) { HIP_RETURN(hipErrorInvalidValue); } *pGraphNode = new hipGraphEventRecordNode(event); @@ -1645,7 +1646,8 @@ hipError_t hipGraphAddEventWaitNode(hipGraphNode_t* pGraphNode, hipGraph_t graph const hipGraphNode_t* pDependencies, size_t numDependencies, hipEvent_t event) { HIP_INIT_API(hipGraphAddEventWaitNode, pGraphNode, graph, pDependencies, numDependencies, event); - if (graph == nullptr || (numDependencies > 0 && pDependencies == nullptr) || event == nullptr) { + if (pGraphNode == nullptr || graph == nullptr || (numDependencies > 0 && pDependencies == nullptr) + || event == nullptr) { HIP_RETURN(hipErrorInvalidValue); } *pGraphNode = new hipGraphEventWaitNode(event);