sroy745 91add85ec4

Fix failing spec decode test (#9054)

2024-10-03 23:07:29 +00:00

..

__init__.py

[Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951)

2024-04-23 08:02:36 +00:00

conftest.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_compatibility.py

Fix failing spec decode test (#9054)

2024-10-03 23:07:29 +00:00

test_eagle_correctness.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_integration_dist_tp2.py

[Speculative Decoding] Test refactor (#8317)

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

test_integration_dist_tp4.py

[Speculative Decoding] Test refactor (#8317)

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

test_integration.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

test_logprobs.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_medusa_correctness.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

test_mlp_correctness.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

test_multistep_correctness.py

[Speculative Decoding] Test refactor (#8317)

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

test_ngram_correctness.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

test_seed.py

[Speculative Decoding] Test refactor (#8317)

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