From 4048789d06b0db5615eb98156a1c67bbe520b5ca Mon Sep 17 00:00:00 2001
From: Ajay <ajay.gunashekar@amd.com>
Date: Mon, 24 Jan 2022 22:41:32 +0000
Subject: [PATCH] SWDEV-317111 - null ptr check for hipGraphGetEdges

Change-Id: I00b7e4d05864b01bfabdf11ea9ba1f2987e4707c


[ROCm/clr commit: 1f00d8c541665f92aebfe019a814be1e02214770]
---
 projects/clr/hipamd/src/hip_graph.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/projects/clr/hipamd/src/hip_graph.cpp b/projects/clr/hipamd/src/hip_graph.cpp
index f9d9a8f986..ba0f09bf1d 100644
--- a/projects/clr/hipamd/src/hip_graph.cpp
+++ b/projects/clr/hipamd/src/hip_graph.cpp
@@ -1187,7 +1187,8 @@ hipError_t hipGraphRemoveDependencies(hipGraph_t graph, const hipGraphNode_t* fr
 hipError_t hipGraphGetEdges(hipGraph_t graph, hipGraphNode_t* from, hipGraphNode_t* to,
                             size_t* numEdges) {
   HIP_INIT_API(hipGraphGetEdges, graph, from, to, numEdges);
-  if (graph == nullptr || numEdges == nullptr) {
+  if (graph == nullptr || numEdges == nullptr ||
+      from == nullptr || to == nullptr) {
     HIP_RETURN(hipErrorInvalidValue);
   }
   const std::vector<std::pair<Node, Node>> edges = graph->GetEdges();