Skip to content

Commit da1e57d

Browse files
author
Jaime Arteaga
committed
Port Fix PI_KERNEL_GROUP_INFO_GLOBAL_WORK_SIZE queries
intel#8769 Signed-off-by: Jaime Arteaga <jaime.a.arteaga.molina@intel.com>
1 parent 611c450 commit da1e57d

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

sycl/plugins/unified_runtime/ur/adapters/level_zero/ur_level_zero_kernel.cpp

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -500,10 +500,13 @@ UR_APIEXPORT ur_result_t UR_APICALL urKernelGetGroupInfo(
500500
// TODO: To revisit after level_zero/issues/262 is resolved
501501
struct {
502502
size_t Arr[3];
503-
} WorkSize = {{Device->ZeDeviceComputeProperties->maxGroupSizeX,
504-
Device->ZeDeviceComputeProperties->maxGroupSizeY,
505-
Device->ZeDeviceComputeProperties->maxGroupSizeZ}};
506-
return ReturnValue(WorkSize);
503+
} GlobalWorkSize = {{(Device->ZeDeviceComputeProperties->maxGroupSizeX *
504+
Device->ZeDeviceComputeProperties->maxGroupCountX),
505+
(Device->ZeDeviceComputeProperties->maxGroupSizeY *
506+
Device->ZeDeviceComputeProperties->maxGroupCountY),
507+
(Device->ZeDeviceComputeProperties->maxGroupSizeZ *
508+
Device->ZeDeviceComputeProperties->maxGroupCountZ)}};
509+
return ReturnValue(GlobalWorkSize);
507510
}
508511
case UR_KERNEL_GROUP_INFO_WORK_GROUP_SIZE: {
509512
// As of right now, L0 is missing API to query kernel and device specific

0 commit comments

Comments
 (0)