From 0d2e0d16212dc9ea3f33d2adf8e150424aa0a39f Mon Sep 17 00:00:00 2001 From: Ioannis Assiouras Date: Tue, 10 Jan 2023 14:28:39 +0000 Subject: [PATCH] SWDEV-374403 - Added input sanity checks for hipLaunchCooperativeKernelMultiDevice and hipExtLaunchMultiKernelMultiDevice Change-Id: Iebef5ae68439d87549fdf9559de51c6c79c64392 --- hipamd/src/hip_module.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/hipamd/src/hip_module.cpp b/hipamd/src/hip_module.cpp index e14b33462d..01adeab472 100644 --- a/hipamd/src/hip_module.cpp +++ b/hipamd/src/hip_module.cpp @@ -508,7 +508,11 @@ hipError_t ihipModuleLaunchCooperativeKernelMultiDevice(hipFunctionLaunchParams* hipError_t result = hipSuccess; result = ihipDeviceGetCount(&numActiveGPUs); - if (numDevices > numActiveGPUs) { + if ((numDevices == 0) || (numDevices > numActiveGPUs)) { + return hipErrorInvalidValue; + } + + if ((flags < 0) || (flags > hipCooperativeLaunchMultiDeviceNoPostSync)) { return hipErrorInvalidValue; }