Add pytorch sampling backend ut (#1425)

2024-09-14 23:15:30 +08:00
parent 30b404ce72
commit 33b54e7c40
2 changed files with 45 additions and 0 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -17,6 +17,7 @@ suites = {
        "test_skip_tokenizer_init.py",
        "test_torch_compile.py",
        "test_triton_attn_backend.py",
+        "test_pytorch_sampling_backend.py",
        "test_update_weights.py",
        "test_vision_openai_server.py",
        "test_server_args.py",