heeju-kim2 2e7796f2cf [Speculative decoding] CUDA graph support (#4295) ...

Co-authored-by: Cade Daniel <edacih@gmail.com>

2024-05-10 17:36:25 +00:00

..

e2e

[Speculative decoding] CUDA graph support (#4295)

2024-05-10 17:36:25 +00:00

__init__.py

[Speculative decoding 3/9] Worker which speculates, scores, and applies rejection sampling (#3103)

2024-03-08 23:32:46 -08:00

test_batch_expansion.py

[Misc] Add pytest marker to opt-out of global test cleanup (#3863)

2024-04-04 21:54:16 -07:00

test_dynamic_spec_decode.py

[Dynamic Spec Decoding] Auto-disable by the running queue size (#4592)

2024-05-08 21:44:00 +00:00

test_metrics.py

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

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

test_multi_step_worker.py

[Misc][Refactor] Introduce ExecuteModelData (#4540)

2024-05-03 17:47:07 -07:00

test_ngram_worker.py

[Misc][Refactor] Introduce ExecuteModelData (#4540)

2024-05-03 17:47:07 -07:00

test_spec_decode_worker.py

[Misc][Refactor] Introduce ExecuteModelData (#4540)

2024-05-03 17:47:07 -07:00

test_utils.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

utils.py

[Misc][Refactor] Introduce ExecuteModelData (#4540)

2024-05-03 17:47:07 -07:00