From f2254e0c34471720f19474cc38cd6ed38a146fdf Mon Sep 17 00:00:00 2001
From: Ajay <ajay.gunashekar@amd.com>
Date: Wed, 2 Mar 2022 19:01:47 +0000
Subject: [PATCH] SWDEV-325545 - hipGraphGetEdges when both from/to are null

Change-Id: Id69d87537a8a7fb7df447cee586137ff1342ce26
---
 hipamd/src/hip_graph.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp
index 3dc9ee7614..9ccf2c2b47 100644
--- a/hipamd/src/hip_graph.cpp
+++ b/hipamd/src/hip_graph.cpp
@@ -1292,7 +1292,8 @@ hipError_t hipGraphGetEdges(hipGraph_t graph, hipGraphNode_t* from, hipGraphNode
                             size_t* numEdges) {
   HIP_INIT_API(hipGraphGetEdges, graph, from, to, numEdges);
   if (graph == nullptr || numEdges == nullptr ||
-      from == nullptr || to == nullptr) {
+      (from == nullptr && to != nullptr) ||
+      (to == nullptr && from != nullptr)) {
     HIP_RETURN(hipErrorInvalidValue);
   }
   const std::vector<std::pair<Node, Node>> edges = graph->GetEdges();