[NATIVECPU] Add UR_DEVICE_INFO_MAX_WORK_GROUP_SIZE for device info

oneapi-src · Oct 2, 2024 · df1d56e · df1d56e
1 parent 7f1332e
commit df1d56e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 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});
@@ -357,7 +360,6 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetInfo(ur_device_handle_t hDevice,
  case UR_DEVICE_INFO_GPU_HW_THREADS_PER_EU:
  case UR_DEVICE_INFO_UUID:
  case UR_DEVICE_INFO_DEVICE_ID:
- case UR_DEVICE_INFO_MAX_NUM_SUB_GROUPS:
  case UR_DEVICE_INFO_SUB_GROUP_INDEPENDENT_FORWARD_PROGRESS:
  case UR_DEVICE_INFO_IL_VERSION:
  case UR_DEVICE_INFO_MAX_WORK_GROUPS_3D:

diff --git a/test/conformance/device/device_adapter_native_cpu.match b/test/conformance/device/device_adapter_native_cpu.match
@@ -11,7 +11,6 @@ urDeviceGetInfoTest.Success/UR_DEVICE_INFO_GLOBAL_MEM_FREE
 urDeviceGetInfoTest.Success/UR_DEVICE_INFO_QUEUE_ON_DEVICE_PROPERTIES
 urDeviceGetInfoTest.Success/UR_DEVICE_INFO_QUEUE_ON_HOST_PROPERTIES
 urDeviceGetInfoTest.Success/UR_DEVICE_INFO_IL_VERSION
-urDeviceGetInfoTest.Success/UR_DEVICE_INFO_MAX_NUM_SUB_GROUPS
 urDeviceGetInfoTest.Success/UR_DEVICE_INFO_SUB_GROUP_INDEPENDENT_FORWARD_PROGRESS
 urDeviceGetInfoTest.Success/UR_DEVICE_INFO_UUID
 urDeviceGetInfoTest.Success/UR_DEVICE_INFO_PCI_ADDRESS