SWDEV-356569 - check for mempool device

- mem pool should be from same device Signed-off-by: sdashmiz <shadi.dashmiz@amd.com> Change-Id: Id06a7f070c019548462c235bff3c0cd972e6eb8a [ROCm/clr commit: 33f02abc41]
2022-09-19 09:55:03 -04:00
parent e649f9393a
commit af7db12908
1 changed files with 6 additions and 0 deletions
@@ -44,6 +44,12 @@ hipError_t hipDeviceSetMemPool(int device, hipMemPool_t mem_pool) {
  if ((mem_pool == nullptr) || (device >= g_devices.size())) {
    HIP_RETURN(hipErrorInvalidValue);
  }
+
+  auto poolDevice = reinterpret_cast<hip::MemoryPool*>(mem_pool)->Device();
+  if (poolDevice->deviceId() != device) {
+    HIP_RETURN(hipErrorInvalidDevice);
+  }
+
  g_devices[device]->SetCurrentMemoryPool(reinterpret_cast<hip::MemoryPool*>(mem_pool));
  HIP_RETURN(hipSuccess);
 }