[NATIVECPU] Add UR_DEVICE_INFO_MAX_WORK_GROUP_SIZE for device info

coldav · Sep 11, 2024 · b3ebaaf · b3ebaaf
1 parent 24a8299
commit b3ebaaf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/source/adapters/native_cpu/device.cpp b/source/adapters/native_cpu/device.cpp
@@ -180,6 +180,9 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetInfo(ur_device_handle_t hDevice,
   case UR_DEVICE_INFO_MAX_WORK_GROUP_SIZE:
     // TODO: provide a mechanism to estimate/configure this.
     return ReturnValue(size_t{2048});
+  case UR_DEVICE_INFO_MAX_NUM_SUB_GROUPS:
+    // Set the max sub groups to be the same as the max work group size.
+    return ReturnValue(uint32_t{2048});
   case UR_DEVICE_INFO_MEM_BASE_ADDR_ALIGN:
     // Imported from level_zero
     return ReturnValue(uint32_t{8});