update dispatch_unquantized_gemm

993c31c3 · zhuwenwen · 4b51e6f1 · 993c31c3
Commit 993c31c3 authored Jan 08, 2026 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

vllm/model_executor/layers/utils.py vllm/model_executor/layers/utils.py +2 -1

No files found.
--- a/vllm/model_executor/layers/utils.py
+++ b/vllm/model_executor/layers/utils.py
@@ -188,7 +188,8 @@ def cpu_unquantized_gemm(layer: torch.nn.Module,

 def dispatch_unquantized_gemm() -> Callable[..., torch.Tensor]:
    if current_platform.is_rocm():
-        return rocm_unquantized_gemm
+        # return rocm_unquantized_gemm
+        return default_unquantized_gemm
    elif current_platform.is_cpu():
        return cpu_unquantized_gemm
    else: