File tree 1 file changed +7
-4
lines changed
sycl/plugins/unified_runtime/ur/adapters/level_zero 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(
500
500
// TODO: To revisit after level_zero/issues/262 is resolved
501
501
struct {
502
502
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);
507
510
}
508
511
case UR_KERNEL_GROUP_INFO_WORK_GROUP_SIZE: {
509
512
// 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