Unverified Commit 9b87a579 authored by Yan Ma's avatar Yan Ma Committed by GitHub
Browse files

[Misc][XPU] Use None as device capacity for XPU (#14932)


Signed-off-by: default avataryan ma <yan.ma@intel.com>
parent b539222d
......@@ -37,10 +37,11 @@ class XPUPlatform(Platform):
return "vllm.attention.backends.ipex_attn.IpexAttnBackend"
@staticmethod
def get_device_capability(device_id: int = 0) -> DeviceCapability:
major, minor, *_ = torch.xpu.get_device_capability(
device_id)['version'].split('.')
return DeviceCapability(major=int(major), minor=int(minor))
def get_device_capability(
device_id: int = 0) -> Optional[DeviceCapability]:
# capacity format differs from cuda's and will cause unexpected
# failure, so use None directly
return None
@staticmethod
def get_device_name(device_id: int = 0) -> str:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment