diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp index e638774f78..8db0fd0e19 100644 --- a/hipamd/src/hip_graph.cpp +++ b/hipamd/src/hip_graph.cpp @@ -1628,7 +1628,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);