torch.xpu.get_device_properties

torch.xpu.get_device_properties(device=None) [исходный код]

Получает свойства устройства. Возвращает _XpuDeviceProperties, содержащий следующие свойства устройства:

• name (str): имя устройства.
• platform_name (str): имя платформы SYCL.
• vendor (str): производитель устройства.
• device_id (int): идентификатор устройства (идентификатор продукта).
• driver_version (str): версия драйвера.
• version (str): версия среды выполнения.
• max_compute_units (int): количество параллельных вычислительных блоков.
• gpu_eu_count (int): количество EU (блоков исполнения).
• max_work_group_size: (int): максимальное количество рабочих элементов, разрешённое в рабочей группе.
• max_num_sub_groups (int): максимальное количество подгрупп, поддерживаемых в рабочей группе.
• memory_clock_rate (int) максимальная тактовая частота глобальной памяти устройства в МГц.
• memory_bus_width (int) максимальная ширина шины между устройством и памятью в битах.
• sub_group_sizes: (list[int]): список поддерживаемых размеров подгрупп.
• local_mem_size (int): объём локальной памяти устройства в байтах, доступный для выделения каждой рабочей группе.
• last_level_cache_size (int): размер кэша памяти последнего уровня устройства в байтах, общего для всех ядер Xe (аналог CUDA L2_cache_size).
• has_fp16 (bool): поддерживается ли тип данных float16.
• has_fp64 (bool): поддерживается ли тип данных float64.
• has_atomic64 (bool): поддерживаются ли атомарные операции с 64-битными данными.
• has_bfloat16_conversions (bool): поддерживаются ли преобразования bfloat16.
• has_subgroup_matrix_multiply_accumulate (bool): поддерживается ли DPAS (Dot Product Accumulate Systolic).
• has_subgroup_matrix_multiply_accumulate_tensor_float32 (bool): поддерживается ли DPAS с входными данными tf32.
• has_subgroup_2d_block_io (bool): поддерживается ли 2D-блочный ввод-вывод для эффективного умножения матриц.
• is_integrated_gpu (bool): является ли устройство встроенным GPU (iGPU).
• total_memory (int): объём глобальной памяти устройства в байтах.
• gpu_subslice_count (int): количество подсрезов.
• architecture (int): идентификатор архитектуры устройства (экспериментальный).
• type (str): тип устройства, например «cpu», «gpu», «accelerator», «host», «unknown».
• uuid (Any): UUID устройства (универсальный уникальный идентификатор), 16 байт.

Параметры:

device (torch.device или int или str) – устройство, свойства которого нужно вернуть.

Возвращает:

свойства устройства

Тип возвращаемого значения:

_XpuDeviceProperties