[ROCm] Split AITER unified attention into its own backend (#25507)

Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
2025-10-06 18:49:23 -04:00
parent 2161efe978
commit f231e5bc21
8 changed files with 325 additions and 301 deletions
--- a/vllm/engine/arg_utils.py
+++ b/vllm/engine/arg_utils.py
@@ -1623,6 +1623,7 @@ class EngineArgs:
            "TREE_ATTN",
            "XFORMERS",
            "ROCM_ATTN",
+            "ROCM_AITER_UNIFIED_ATTN",
        ]
        if (
            envs.is_set("VLLM_ATTENTION_BACKEND")