torch.xpu.get_device_capability

torch.xpu.get_device_capability(device=None)

获取设备的 xpu 能力。

参数:

device (torch.device 或 int 或 str, 可选) – 要返回设备能力的设备。如果此参数为负整数，则此函数无效。如果 device 为 None (默认)，则使用当前设备，由 current_device() 指定。

返回:

设备的 xpu 能力字典

返回类型:

dict[str, Any]