Lianmin Zheng dafb6a5266 [Fix] Fix the style of test_large_max_new_tokens.py (#1638)

2024-10-11 16:05:58 -07:00

..

models

Fix the correctness test in bench_latency.py when tp > 1 and test_generation_models.py (#1631)

2024-10-11 05:03:20 -07:00

sampling/penaltylib

[CI] Fix CI (#1217)

2024-08-26 02:56:42 +00:00

run_suite.py

[engine] support async and streaming (#1614)

2024-10-11 15:26:25 -07:00

test_bench_latency.py

Enable torch.compile for triton backend (#1422)

2024-09-14 15:38:37 -07:00

test_bench_serving.py

[Minor, Performance] Use torch.argmax for greedy sampling (#1589)

2024-10-06 13:15:05 -07:00

test_chunked_prefill.py

Fix chunked prefill condition (#1594)

2024-10-07 06:34:14 +00:00

test_create_kvindices.py

Organize Attention Backends (#1547)

2024-09-30 15:54:18 -07:00

test_data_parallelism.py

Add a unit test for data parallelism (#1489)

2024-09-22 02:21:05 -07:00

test_embedding_openai_server.py

improve the threshold and ports in tests (#1215)

2024-08-25 19:02:08 -07:00

test_eval_accuracy_large_chunked_prefill.py

improve the threshold and ports in tests (#1215)

2024-08-25 19:02:08 -07:00

test_eval_accuracy_large_mixed_chunked_prefill.py

improve the threshold and ports in tests (#1215)

2024-08-25 19:02:08 -07:00

test_eval_accuracy_large.py

improve the threshold and ports in tests (#1215)

2024-08-25 19:02:08 -07:00

test_eval_accuracy_mini.py

Add torchao quant (int4/int8/fp8) to llama models (#1341)

2024-09-09 05:32:41 -07:00

test_json_constrained.py

Fix constrained decoding (#1634)

2024-10-11 06:26:20 -07:00

test_large_max_new_tokens.py

[Fix] Fix the style of test_large_max_new_tokens.py (#1638)

2024-10-11 16:05:58 -07:00

test_mla_fp8.py

minor: add mla fp8 test (#1494)

2024-09-23 20:40:17 +08:00

test_mla.py

Add MLA gsm8k eval (#1484)

2024-09-21 11:16:13 +08:00

test_models_from_modelscope.py

Improve process creation (#1534)

2024-09-29 02:36:12 -07:00

test_moe_eval_accuracy_large.py

Enable torch.compile for triton backend (#1422)

2024-09-14 15:38:37 -07:00

test_nightly_gsm8k_eval.py

minor: add quant eval compared with base (#1475)

2024-09-20 01:57:19 +08:00

test_openai_server.py

[Fix] Fix all the Huggingface paths (#1553)

2024-10-02 10:12:07 -07:00

test_pytorch_sampling_backend.py

Test consistency for single and batch seperately (#1590)

2024-10-06 22:02:27 +00:00

test_server_args.py

[Minor] Many cleanup (#1357)

2024-09-09 04:14:11 -07:00

test_skip_tokenizer_init.py

improve the threshold and ports in tests (#1215)

2024-08-25 19:02:08 -07:00

test_srt_endpoint.py

[Fix] Fix logprob and normalized_logprob (#1428)

2024-09-15 06:36:06 -07:00

test_srt_engine.py

[engine] support async and streaming (#1614)

2024-10-11 15:26:25 -07:00

test_torch_compile.py

Add torchao quant (int4/int8/fp8) to llama models (#1341)

2024-09-09 05:32:41 -07:00

test_torchao.py

[Fix] Fix logprob and normalized_logprob (#1428)

2024-09-15 06:36:06 -07:00

test_triton_attention_kernels.py

Organize Attention Backends (#1547)

2024-09-30 15:54:18 -07:00

test_triton_attn_backend.py

[Minor] Improve the style and fix flaky tests (#1584)

2024-10-06 00:10:48 -07:00

test_update_weights.py

Better unit tests for adding a new model (#1488)

2024-09-22 01:50:37 -07:00

test_vision_openai_server.py

[Feat] Add modalities for vision server when handling pixel values for llava (#1346)

2024-09-09 02:07:34 -07:00