Lily Liu c6bd70d772

[SpecDec][Misc] Cleanup, remove bonus token logic. (#8701)

2024-09-22 12:34:14 -07: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

[Speculative Decoding] Test refactor (#8317)

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

test_compatibility.py

[Speculative decoding][Re-take] Enable TP>1 speculative decoding (#4840)

2024-05-16 00:53:51 -07:00

test_eagle_correctness.py

[Speculative Decoding] Test refactor (#8317)

2024-09-11 14:07:34 -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

[Speculative Decoding] Test refactor (#8317)

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

test_logprobs.py

[Speculative Decoding] Test refactor (#8317)

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

test_medusa_correctness.py

[SpecDec][Misc] Cleanup, remove bonus token logic. (#8701)

2024-09-22 12:34:14 -07:00

test_mlp_correctness.py

[Speculative Decoding] Test refactor (#8317)

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

test_multistep_correctness.py

[Speculative Decoding] Test refactor (#8317)

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

test_ngram_correctness.py

[Speculative Decoding] Test refactor (#8317)

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

test_seed.py

[Speculative Decoding] Test refactor (#8317)

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