We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5b1895e commit d9da93fCopy full SHA for d9da93f
vllm/platforms/rocm.py
@@ -98,7 +98,7 @@ def wrapper(*args, **kwargs):
98
@cache
99
def on_mi3xx() -> bool:
100
GPU_ARCH = torch.cuda.get_device_properties("cuda").gcnArchName
101
- return any(arch in GPU_ARCH for arch in ["gfx90a", "gfx942"])
+ return any(arch in GPU_ARCH for arch in ["gfx942", "gfx950"])
102
103
104
0 commit comments