Fix Triton decode kernel & ut (#1819)

2024-10-28 01:54:38 +08:00
parent 51c81e339b
commit c77762d57f
5 changed files with 218 additions and 42 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -26,7 +26,8 @@ suites = {
        "test_srt_endpoint.py",
        "test_torch_compile.py",
        "test_torchao.py",
-        "test_triton_attn_backend.py",
+        "test_triton_attention_kernels.py",
+        "test_triton_attention_backend.py",
        "test_update_weights.py",
        "test_vision_openai_server.py",
    ],