add use_sparse

23f39c5b · zhuwenwen · 70764a8f · 23f39c5b
Commit 23f39c5b authored Nov 14, 2025 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

vllm/platforms/rocm.py vllm/platforms/rocm.py +3 -3

No files found.
--- a/vllm/platforms/rocm.py
+++ b/vllm/platforms/rocm.py
@@ -215,9 +215,9 @@ class RocmPlatform(Platform):
    def get_attn_backend_cls(cls, selected_backend, head_size, dtype,
                             kv_cache_dtype, block_size, use_v1, use_mla,
                             has_sink, use_sparse) -> str:
-        if use_sparse:
-            raise NotImplementedError(
-                "Sparse Attention is not supported on ROCm.")
+        # if use_sparse:
+        #     raise NotImplementedError(
+        #         "Sparse Attention is not supported on ROCm.")
        if use_mla:
            if not use_v1:
                raise RuntimeError(