Skip to content

Commit 9b87a57

Browse files
authored
[Misc][XPU] Use None as device capacity for XPU (#14932)
Signed-off-by: yan ma <[email protected]>
1 parent b539222 commit 9b87a57

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

vllm/platforms/xpu.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,11 @@ def get_attn_backend_cls(cls, selected_backend: _Backend, head_size: int,
3737
return "vllm.attention.backends.ipex_attn.IpexAttnBackend"
3838

3939
@staticmethod
40-
def get_device_capability(device_id: int = 0) -> DeviceCapability:
41-
major, minor, *_ = torch.xpu.get_device_capability(
42-
device_id)['version'].split('.')
43-
return DeviceCapability(major=int(major), minor=int(minor))
40+
def get_device_capability(
41+
device_id: int = 0) -> Optional[DeviceCapability]:
42+
# capacity format differs from cuda's and will cause unexpected
43+
# failure, so use None directly
44+
return None
4445

4546
@staticmethod
4647
def get_device_name(device_id: int = 0) -> str:

0 commit comments

Comments
 (0)