diff --git a/unified-runtime/source/adapters/level_zero/usm.cpp b/unified-runtime/source/adapters/level_zero/usm.cpp index 4694ee5411ceb..58330a223bfa6 100644 --- a/unified-runtime/source/adapters/level_zero/usm.cpp +++ b/unified-runtime/source/adapters/level_zero/usm.cpp @@ -411,6 +411,9 @@ ur_result_t urUSMGetMemAllocInfo( case ZE_MEMORY_TYPE_SHARED: MemAllocaType = UR_USM_TYPE_SHARED; break; + case ZE_MEMORY_TYPE_HOST_IMPORTED: + MemAllocaType = UR_USM_TYPE_HOST; + break; default: UR_LOG(ERR, "urUSMGetMemAllocInfo: unexpected usm memory type"); return UR_RESULT_ERROR_INVALID_VALUE; diff --git a/unified-runtime/source/adapters/level_zero/v2/usm.cpp b/unified-runtime/source/adapters/level_zero/v2/usm.cpp index 337484808c791..770d2488f6220 100644 --- a/unified-runtime/source/adapters/level_zero/v2/usm.cpp +++ b/unified-runtime/source/adapters/level_zero/v2/usm.cpp @@ -809,6 +809,9 @@ ur_result_t urUSMGetMemAllocInfo( case ZE_MEMORY_TYPE_SHARED: memAllocType = UR_USM_TYPE_SHARED; break; + case ZE_MEMORY_TYPE_HOST_IMPORTED: + memAllocType = UR_USM_TYPE_HOST; + break; default: UR_LOG(ERR, "urUSMGetMemAllocInfo: unexpected usm memory type"); return UR_RESULT_ERROR_INVALID_VALUE;