From 8fad1177e36af8549a5d94ef6692f7677c37ee78 Mon Sep 17 00:00:00 2001 From: Ajay Date: Mon, 14 Mar 2022 21:08:25 +0000 Subject: [PATCH] SWDEV-315756 - hipGraphAddMemcpyNodeFromSymbol issues Change-Id: Ibf9504454830b63caf6048aa53d726116da6d713 --- hipamd/src/hip_graph.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp index e39db5bdcf..97f3a33bd1 100644 --- a/hipamd/src/hip_graph.cpp +++ b/hipamd/src/hip_graph.cpp @@ -1485,6 +1485,11 @@ hipError_t hipGraphAddMemcpyNodeFromSymbol(hipGraphNode_t* pGraphNode, hipGraph_ size_t count, size_t offset, hipMemcpyKind kind) { HIP_INIT_API(hipGraphAddMemcpyNodeFromSymbol, pGraphNode, graph, pDependencies, numDependencies, dst, symbol, count, offset, kind); + if (graph == nullptr || pGraphNode == nullptr || pDependencies == nullptr || + dst == nullptr || !ihipGraph::isGraphValid(graph)) { + HIP_RETURN(hipErrorInvalidValue); + } + size_t sym_size = 0; hipDeviceptr_t device_ptr = nullptr;