[HIP] Support managed allocations in USM free

In ROCm 5.7.1 some USM allocations are reported as managed, these are freed in the same way as device allocations.
oneapi-src · Jan 29, 2024 · 9c68768 · 9c68768
1 parent e79993d
commit 9c68768
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/source/adapters/hip/usm.cpp b/source/adapters/hip/usm.cpp
@@ -78,9 +78,10 @@ USMFreeImpl([[maybe_unused]] ur_context_handle_t hContext, void *pMem) {
 #else
  const auto Type = hipPointerAttributeType.memoryType;
 #endif
- UR_ASSERT(Type == hipMemoryTypeDevice || Type == hipMemoryTypeHost,
+ UR_ASSERT(Type == hipMemoryTypeDevice || Type == hipMemoryTypeHost ||
+ Type == hipMemoryTypeManaged,
  UR_RESULT_ERROR_INVALID_MEM_OBJECT);
- if (Type == hipMemoryTypeDevice) {
+ if (Type == hipMemoryTypeDevice || Type == hipMemoryTypeManaged) {
  UR_CHECK_ERROR(hipFree(pMem));
  }
  if (Type == hipMemoryTypeHost) {