From ea9847c1fd438f5fb570974841fe594ef1c12043 Mon Sep 17 00:00:00 2001
From: Sourabh Betigeri <sourabh.betigeri@amd.com>
Date: Mon, 23 May 2022 17:11:28 -0700
Subject: [PATCH] SWDEV-320640 - Fixes to return right error codes in
 hipStreamUpdateCaptureDependencies() for negative scenarios

Change-Id: I3c4c54f8955fa9bb17eb595544005fe9f0e8a5f0


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

diff --git a/projects/clr/hipamd/src/hip_graph.cpp b/projects/clr/hipamd/src/hip_graph.cpp
index c16b76d9d0..a0bffed2f7 100644
--- a/projects/clr/hipamd/src/hip_graph.cpp
+++ b/projects/clr/hipamd/src/hip_graph.cpp
@@ -1321,7 +1321,10 @@ hipError_t hipStreamUpdateCaptureDependencies(hipStream_t stream, hipGraphNode_t
   if (s->GetCaptureStatus() == hipStreamCaptureStatusNone) {
     HIP_RETURN(hipErrorIllegalState);
   }
-  if (numDependencies > 0 && dependencies == nullptr) {
+  if (numDependencies > 0 && dependencies == nullptr ||
+      (numDependencies > s->GetLastCapturedNodes().size()) ||
+      (flags != 0 && (flags != hipStreamAddCaptureDependencies ||
+       flags != hipStreamSetCaptureDependencies))) {
     HIP_RETURN(hipErrorInvalidValue);
   }
   std::vector<hipGraphNode_t> depNodes;