@@ -512,10 +512,11 @@ ur_result_t urKernelGetGroupInfo(
512
512
ZE_CALL_NOCHECK (zeKernelGetProperties, (zeDevice, &kernelProperties));
513
513
if (zeResult == ZE_RESULT_SUCCESS &&
514
514
workGroupProperties.maxGroupSize != 0 ) {
515
- return returnValue (workGroupProperties.maxGroupSize );
515
+ // Specification states this returns a size_t.
516
+ return returnValue (size_t {workGroupProperties.maxGroupSize });
516
517
}
517
518
return returnValue (
518
- uint64_t {hDevice->ZeDeviceComputeProperties ->maxTotalGroupSize });
519
+ size_t {hDevice->ZeDeviceComputeProperties ->maxTotalGroupSize });
519
520
}
520
521
case UR_KERNEL_GROUP_INFO_COMPILE_WORK_GROUP_SIZE: {
521
522
auto props = hKernel->getProperties (hDevice);
@@ -527,15 +528,15 @@ ur_result_t urKernelGetGroupInfo(
527
528
}
528
529
case UR_KERNEL_GROUP_INFO_LOCAL_MEM_SIZE: {
529
530
auto props = hKernel->getProperties (hDevice);
530
- return returnValue (uint32_t {props.localMemSize });
531
+ return returnValue (size_t {props.localMemSize });
531
532
}
532
533
case UR_KERNEL_GROUP_INFO_PREFERRED_WORK_GROUP_SIZE_MULTIPLE: {
533
534
return returnValue (
534
535
size_t {hDevice->ZeDeviceProperties ->physicalEUSimdWidth });
535
536
}
536
537
case UR_KERNEL_GROUP_INFO_PRIVATE_MEM_SIZE: {
537
538
auto props = hKernel->getProperties (hDevice);
538
- return returnValue (uint32_t {props.privateMemSize });
539
+ return returnValue (size_t {props.privateMemSize });
539
540
}
540
541
case UR_KERNEL_GROUP_INFO_COMPILE_MAX_WORK_GROUP_SIZE:
541
542
case UR_KERNEL_GROUP_INFO_COMPILE_MAX_LINEAR_WORK_GROUP_SIZE:
0 commit comments