From f4f377bcaaa3183b332bec10b89e2e725d08a108 Mon Sep 17 00:00:00 2001
From: Ioannis Assiouras <Ioannis.Assiouras@amd.com>
Date: Tue, 20 Dec 2022 15:49:15 +0000
Subject: [PATCH] SWDEV-374402 - Added input stream check for
 hipExtLaunchKernel

Change-Id: I42915c3b9b10b4a4a5b3ae74b6a2dccc8fddbf7f
---
 hipamd/src/hip_module.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hipamd/src/hip_module.cpp b/hipamd/src/hip_module.cpp
index e1c801b81f..a3267249a5 100644
--- a/hipamd/src/hip_module.cpp
+++ b/hipamd/src/hip_module.cpp
@@ -496,6 +496,11 @@ extern "C" hipError_t hipExtLaunchKernel(const void* hostFunction, dim3 gridDim,
                                          hipEvent_t startEvent, hipEvent_t stopEvent, int flags) {
   HIP_INIT_API(hipExtLaunchKernel, hostFunction, gridDim, blockDim, args, sharedMemBytes,
                stream, startEvent, stopEvent, flags);
+
+  if (!hip::isValid(stream)) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
+
   STREAM_CAPTURE(hipExtLaunchKernel, stream, hostFunction, gridDim, blockDim, args, sharedMemBytes,
                  startEvent, stopEvent, flags);
   HIP_RETURN(ihipLaunchKernel(hostFunction, gridDim, blockDim, args, sharedMemBytes, stream,