From 522b62d87a4a1aae699f322af38947a404f4d3c1 Mon Sep 17 00:00:00 2001
From: Ajay <ajay.gunashekar@amd.com>
Date: Wed, 2 Feb 2022 23:27:28 +0000
Subject: [PATCH] SWDEV-316465 - hipStreamGetCaptureInfo_v2 pointer checks

Change-Id: I219e16909ddd3fa99cd9a2faefc0f35237ee3b08


[ROCm/clr commit: 57d0acb6d7c9054b35dc6e6ba0ccd02c0740020a]
---
 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 a851ee3348..3f35c1321a 100644
--- a/projects/clr/hipamd/src/hip_graph.cpp
+++ b/projects/clr/hipamd/src/hip_graph.cpp
@@ -1156,8 +1156,11 @@ hipError_t hipStreamGetCaptureInfo_v2(hipStream_t stream, hipStreamCaptureStatus
                                       size_t* numDependencies_out) {
   HIP_INIT_API(hipStreamGetCaptureInfo_v2, stream, captureStatus_out, id_out, graph_out,
                dependencies_out, numDependencies_out);
+  if (captureStatus_out == nullptr) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
   if (stream == nullptr) {
-    HIP_RETURN(hipErrorStreamCaptureImplicit);
+    HIP_RETURN(hipErrorUnknown);
   }
   if (!hip::isValid(stream)) {
     HIP_RETURN(hipErrorInvalidValue);