diff --git a/projects/clr/hipamd/src/hip_mempool.cpp b/projects/clr/hipamd/src/hip_mempool.cpp index 32e1c758f5..312becf4db 100644 --- a/projects/clr/hipamd/src/hip_mempool.cpp +++ b/projects/clr/hipamd/src/hip_mempool.cpp @@ -28,9 +28,12 @@ extern hipError_t ihipFree(void* ptr); // ================================================================================================ hipError_t hipDeviceGetDefaultMemPool(hipMemPool_t* mem_pool, int device) { HIP_INIT_API(hipDeviceGetDefaultMemPool, mem_pool, device); - if ((mem_pool == nullptr) || (device >= g_devices.size())) { + if (mem_pool == nullptr) { HIP_RETURN(hipErrorInvalidValue); } + if (device < 0 || device >= g_devices.size()) { + HIP_RETURN(hipErrorInvalidDevice); + } *mem_pool = reinterpret_cast(g_devices[device]->GetDefaultMemoryPool()); HIP_RETURN(hipSuccess); }