Merge pull request oneapi-src#1486 from nrspruit/fix_memfree_report

[L0] Fix DeviceInfo global mem free to report unsupported given MemCount==0
kbenzie · Apr 8, 2024 · 65af257 · 65af257
1 parent 1ee81f4
commit 65af257
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/source/adapters/level_zero/device.cpp b/source/adapters/level_zero/device.cpp
@@ -718,7 +718,11 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetInfo(
         }
       }
     }
-    return ReturnValue(std::min(GlobalMemSize, FreeMemory));
+    if (MemCount > 0) {
+      return ReturnValue(std::min(GlobalMemSize, FreeMemory));
+    } else {
+      return UR_RESULT_ERROR_UNSUPPORTED_FEATURE;
+    }
   }
   case UR_DEVICE_INFO_MEMORY_CLOCK_RATE: {
     // If there are not any memory modules then return 0.