Merge pull request #1486 from nrspruit/fix_memfree_report

aarongreig · web-flow · commit 065bf2dd97b5 · 2024-04-05T14:26:59.000+01:00
[L0] Fix DeviceInfo global mem free to report unsupported given MemCount==0
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.

Original file line number	Diff line number	Diff line change
`@@ -718,7 +718,11 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetInfo(`
`718`	`718`	`}`
`719`	`719`	`}`
`720`	`720`	`}`
`721`		`- return ReturnValue(std::min(GlobalMemSize, FreeMemory));`
	`721`	`+ if (MemCount > 0) {`
	`722`	`+ return ReturnValue(std::min(GlobalMemSize, FreeMemory));`
	`723`	`+ } else {`
	`724`	`+ return UR_RESULT_ERROR_UNSUPPORTED_FEATURE;`
	`725`	`+ }`
`722`	`726`	`}`
`723`	`727`	`case UR_DEVICE_INFO_MEMORY_CLOCK_RATE: {`
`724`	`728`	`// If there are not any memory modules then return 0.`