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; }