SWDEV-328556: check for wait event node type

Signed-off-by: sdashmiz <shadi.dashmiz@amd.com>
Change-Id: I4b70cb576df1ab31503d9089591026bebe4df16a
This commit is contained in:
sdashmiz
2022-03-21 11:01:22 -04:00
committed by Shadi Dashmiz
parent 6a48e430d0
commit 610aa33a8b
+2 -1
View File
@@ -1663,7 +1663,8 @@ hipError_t hipGraphAddEventWaitNode(hipGraphNode_t* pGraphNode, hipGraph_t graph
hipError_t hipGraphEventWaitNodeGetEvent(hipGraphNode_t node, hipEvent_t* event_out) {
HIP_INIT_API(hipGraphEventWaitNodeGetEvent, node, event_out);
if (node == nullptr || event_out == nullptr) {
if (node == nullptr || event_out == nullptr ||
node->GetType() != hipGraphNodeTypeWaitEvent) {
HIP_RETURN(hipErrorInvalidValue);
}
reinterpret_cast<hipGraphEventWaitNode*>(node)->GetParams(event_out);