File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed
sycl/plugins/unified_runtime/ur/adapters/level_zero Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments