Skip to content

Commit cfd7388

Browse files
committed
fix is_hip
1 parent 64f51a5 commit cfd7388

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ def reset(self):
423423

424424
@lru_cache(maxsize=None)
425425
def is_mi250() -> bool:
426-
if not is_hip() or not torch.cuda.is_available():
426+
if not current_platform.is_rocm() or not torch.cuda.is_available():
427427
return False
428428
archName = torch.cuda.get_device_properties('cuda').gcnArchName
429429
return (archName is not None) and \

0 commit comments

Comments
 (0)