Lianmin Zheng 3f0fe08d37 Let ModelRunner take InputMetadata as input, instead of ScheduleBatch (#1541)

2024-09-29 20:28:45 -07:00

..

models

Let ModelRunner take InputMetadata as input, instead of ScheduleBatch (#1541)

2024-09-29 20:28:45 -07:00

sampling/penaltylib

[CI] Fix CI (#1217)

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

run_suite.py

Let ModelRunner take InputMetadata as input, instead of ScheduleBatch (#1541)

2024-09-29 20:28:45 -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

Fix oom issues with fp8 for llama (#1454)

2024-09-18 03:45:19 -07:00

test_chunked_prefill.py

Fix oom issues with fp8 for llama (#1454)

2024-09-18 03:45:19 -07:00

test_create_kvindices.py

Refactor attention backend (#1381)

2024-09-11 11:44:26 -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] Fix logprob and normalized_logprob (#1428)

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

test_large_max_new_tokens.py

[API, Feature] Support response prefill for openai API (#1490)

2024-09-22 06:46:17 -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

[API, Feature] Support response prefill for openai API (#1490)

2024-09-22 06:46:17 -07:00

test_pytorch_sampling_backend.py

Add pytorch sampling backend ut (#1425)

2024-09-15 01:15:30 +10: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_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

[Fix] Fix logprob and normalized_logprob (#1428)

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

test_triton_attn_backend.py

Improve process creation (#1534)

2024-09-29 02:36:12 -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