sroy745 f6778620a9

Disable spec-decode + chunked-prefill for draft models with tensor parallelism > 1 (#10136) ...

Signed-off-by: Sourashis Roy <sroy@roblox.com>

2024-11-08 15:56:18 +00:00

..

e2e

Disable spec-decode + chunked-prefill for draft models with tensor parallelism > 1 (#10136)

2024-11-08 15:56:18 +00:00

__init__.py

…

test_batch_expansion.py

[BugFix] Fix use of per-request seed with pipeline parallel (#6698)

2024-07-30 10:40:08 -07:00

test_dynamic_spec_decode.py

…

test_metrics.py

…

test_multi_step_worker.py

[Bugfix][SpecDecode] kv corruption with bonus tokens in spec decode (#9730)

2024-11-06 01:45:45 +00:00

test_ngram_worker.py

[Feature] [Spec decode]: Combine chunked prefill with speculative decoding (#9291)

2024-11-07 08:15:14 -08:00

test_scorer.py

[Feature] [Spec decode]: Combine chunked prefill with speculative decoding (#9291)

2024-11-07 08:15:14 -08:00

test_spec_decode_worker.py

[Feature] [Spec decode]: Combine chunked prefill with speculative decoding (#9291)

2024-11-07 08:15:14 -08:00

test_utils.py

[Bugfix] Unify rank computation across regular decoding and speculative decoding (#7899)

2024-08-28 22:18:13 -07:00

utils.py

[Feature] [Spec decode]: Combine chunked prefill with speculative decoding (#9291)

2024-11-07 08:15:14 -08:00