[L0 v2] fix error reporting for urUSM*Alloc

match the spec behavior
oneapi-src · Nov 14, 2024 · d31979b · d31979b
1 parent 50ef8c9
commit d31979b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/source/adapters/level_zero/v2/usm.cpp b/source/adapters/level_zero/v2/usm.cpp
@@ -22,8 +22,12 @@
 namespace umf {
 ur_result_t getProviderNativeError(const char *providerName,
                                    int32_t nativeError) {
-  if (strcmp(providerName, "Level Zero") == 0) {
-    return ze2urResult(static_cast<ze_result_t>(nativeError));
+  if (strcmp(providerName, "LEVEL_ZERO") == 0) {
+    auto zeResult = static_cast<ze_result_t>(nativeError);
+    if (zeResult == ZE_RESULT_ERROR_UNSUPPORTED_SIZE) {
+      return UR_RESULT_ERROR_INVALID_USM_SIZE;
+    }
+    return ze2urResult(zeResult);
   }
 
   return UR_RESULT_ERROR_UNKNOWN;

diff --git a/test/conformance/usm/usm_adapter_level_zero_v2.match b/test/conformance/usm/usm_adapter_level_zero_v2.match
@@ -1,4 +1 @@
-urUSMDeviceAllocTest.InvalidUSMSize/*
 urUSMGetMemAllocInfoTest.Success/*___UR_USM_ALLOC_INFO_POOL
-urUSMHostAllocTest.InvalidUSMSize/*
-urUSMSharedAllocTest.InvalidUSMSize/*