From df1d56e02af6b7e33e80bb17edd22483041f5865 Mon Sep 17 00:00:00 2001 From: Colin Davidson Date: Wed, 11 Sep 2024 16:47:02 +0100 Subject: [PATCH] [NATIVECPU] Add UR_DEVICE_INFO_MAX_WORK_GROUP_SIZE for device info --- source/adapters/native_cpu/device.cpp | 4 +++- test/conformance/device/device_adapter_native_cpu.match | 1 - 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/source/adapters/native_cpu/device.cpp b/source/adapters/native_cpu/device.cpp index 15dbed75c9..9f2c05a48d 100644 --- 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 index b2f88f3bc8..2129478fb8 100644 --- 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