[Bugfix] Fix broken kernel test due to missing rename for v1 Triton backend (#15282)

Signed-off-by: Isotr0py <2037008807@qq.com>
2025-03-21 19:42:06 +08:00
parent 91ca929dc7
commit 8afcd0f633
2 changed files with 2 additions and 2 deletions
--- a/tests/kernels/test_attention_selector.py
+++ b/tests/kernels/test_attention_selector.py
@@ -49,7 +49,7 @@ def test_env(
                       RocmPlatform()):
                backend = get_attn_backend(16, torch.float16, torch.float16,
                                           16, False)
-            EXPECTED = "ROCM_ATTN_VLLM_V1" if use_v1 else "ROCM_FLASH"
+            EXPECTED = "TRITON_ATTN_VLLM_V1" if use_v1 else "ROCM_FLASH"
            assert backend.get_name() == EXPECTED
        elif device == "openvino":
            with patch("vllm.attention.selector.current_platform",