From c328f3b15f2a8af9ba4ea9271f8f02132e464fff Mon Sep 17 00:00:00 2001
From: pghafari <pghafari@amd.com>
Date: Mon, 19 Sep 2022 07:36:40 -0400
Subject: [PATCH] SWDEV-356554 - updated hipDeviceGetDefaultMemPool

updated the error check/messages

Change-Id: I518d497a82a14862d5168cc423aeaa20075336b8


[ROCm/clr commit: 42b5a6d5af3e110a2779e227069a8d9912de589b]
---
 projects/clr/hipamd/src/hip_mempool.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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<hipMemPool_t>(g_devices[device]->GetDefaultMemoryPool());
   HIP_RETURN(hipSuccess);
 }