Merge pull request #1193 from npmiller/managed-alloc

[HIP] Support managed allocations in USM free
oneapi-src · Feb 2, 2024 · 2974c52 · 2974c52
2 parents 8c604b0 + 9c68768
commit 2974c52
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) {