diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp index af725f57ee..873ecaee22 100644 --- a/hipamd/src/hip_graph.cpp +++ b/hipamd/src/hip_graph.cpp @@ -1134,10 +1134,13 @@ hipError_t hipGraphExecKernelNodeSetParams(hipGraphExec_t hGraphExec, hipGraphNo hipError_t hipGraphChildGraphNodeGetGraph(hipGraphNode_t node, hipGraph_t* pGraph) { HIP_INIT_API(hipGraphChildGraphNodeGetGraph, node, pGraph); - if (node == nullptr) { + if (node == nullptr || pGraph == nullptr) { HIP_RETURN(hipErrorInvalidValue); } *pGraph = reinterpret_cast(node)->GetChildGraph(); + if (pGraph == nullptr) { + HIP_RETURN(hipErrorInvalidValue); + } HIP_RETURN(hipSuccess); }