Skip to content

Commit d9da93f

Browse files
committed
fix on_mi3xx
Signed-off-by: charlifu <[email protected]>
1 parent 5b1895e commit d9da93f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/platforms/rocm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def wrapper(*args, **kwargs):
9898
@cache
9999
def on_mi3xx() -> bool:
100100
GPU_ARCH = torch.cuda.get_device_properties("cuda").gcnArchName
101-
return any(arch in GPU_ARCH for arch in ["gfx90a", "gfx942"])
101+
return any(arch in GPU_ARCH for arch in ["gfx942", "gfx950"])
102102

103103

104104
@cache

0 commit comments

Comments
 (0)