dpctl.SyclDevice.sub_group_sizes

SyclDevice.sub_group_sizes

Returns list of supported sub-group sizes for this device.

Example:

>>> import dpctl
>>> dev = dpctl.select_cpu_device()
>>> dev.sub_group_sizes
[4, 8, 16, 32, 64]

Returns:

List of supported sub-group sizes.

Return type:

List[int]