torch.xpu.get_device_capability — PyTorch 2.7 documentation (original) (raw)

torch.xpu.get_device_capability(device=None)[source][source]¶

Get the xpu capability of a device.

Parameters

device (torch.device or int or str, optional) – device for which to return the device capability. This function is a no-op if this argument is a negative integer. It uses the current device, given bycurrent_device(), if device is None(default).

Returns

the xpu capability dictionary of the device

Return type

Dict[str, Any]

• torch.xpu.get_device_capability
  • get_device_capability()

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.