SWDEV-315756 - hipGraphAddMemcpyNodeFromSymbol issues

Change-Id: Ibf9504454830b63caf6048aa53d726116da6d713
2022-03-14 21:08:25 +00:00
parent f8f8b0fd95
commit 8fad1177e3
1 changed files with 5 additions and 0 deletions
@@ -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;