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 newGraphNodes; hGraph->LevelOrder(newGraphNodes); std::vector& oldGraphExecNodes = hGraphExec->GetNodes();