Re-organize CI tests (#1052)

2024-08-12 03:39:01 -07:00
parent 0c1c72a0b4
commit c877292cc1
9 changed files with 148 additions and 22 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -12,6 +12,7 @@ suites = {
        "test_openai_server.py",
        "test_skip_tokenizer_init.py",
        "test_torch_compile.py",
+        "test_triton_attn_backend.py",
        "test_vision_openai_server.py",
        "test_large_max_new_tokens.py",
        "models/test_generation_models.py",