From 27d1a6ea9424fd059bf6981543b581755053c8e1 Mon Sep 17 00:00:00 2001
From: Ajay <ajay.gunashekar@amd.com>
Date: Mon, 14 Mar 2022 21:08:25 +0000
Subject: [PATCH] SWDEV-315756 - hipGraphAddMemcpyNodeFromSymbol issues

Change-Id: Ibf9504454830b63caf6048aa53d726116da6d713


[ROCm/clr commit: 8fad1177e36af8549a5d94ef6692f7677c37ee78]
---
 projects/clr/hipamd/src/hip_graph.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/projects/clr/hipamd/src/hip_graph.cpp b/projects/clr/hipamd/src/hip_graph.cpp
index e39db5bdcf..97f3a33bd1 100644
--- a/projects/clr/hipamd/src/hip_graph.cpp
+++ b/projects/clr/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;