Merge pull request #2330 from igchor/v2_usm_error

[L0 v2] fix error reporting for urUSM*Alloc
oneapi-src · Nov 25, 2024 · c7086f7 · c7086f7
2 parents e9d94b1 + d31979b
commit c7086f7
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/*