From 596b16da5a87d2aa1ae846ea64d30ae0bc6a721f Mon Sep 17 00:00:00 2001
From: Anusha GodavarthySurya <anusha.godavarthysurya@amd.com>
Date: Wed, 30 Nov 2022 04:35:01 +0000
Subject: [PATCH] SWDEV-325711 - Added graph API hipGraphUpload

Change-Id: I7e4c8c997a5ef940000d599ded9d317d62b8826b
---
 hipamd/src/hip_graph.cpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp
index 3ddb23b549..55a1e7914e 100644
--- a/hipamd/src/hip_graph.cpp
+++ b/hipamd/src/hip_graph.cpp
@@ -2200,5 +2200,10 @@ hipError_t hipGraphNodeGetEnabled(hipGraphExec_t hGraphExec, hipGraphNode_t hNod
 
 hipError_t hipGraphUpload(hipGraphExec_t graphExec, hipStream_t stream) {
   HIP_INIT_API(hipGraphUpload, graphExec, stream);
-  HIP_RETURN(hipErrorNotSupported);
+  if (graphExec == nullptr || !hip::isValid(stream)) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
+  // TODO: stream is known before launch, do preperatory work with graph optimizations. pre-allocate
+  // memory for memAlloc nodes if any when support is added with mempool feature
+  HIP_RETURN(hipSuccess);
 }