| .. |
|
__init__.py
|
[TRTLLM-10321][feat] Support different KV cache layout for one-model spec dec (#10502)
|
2026-02-10 05:16:02 +08:00 |
|
auto_heuristic.py
|
[None][feat] Clean up ngram auto mode, add max_concurrency to configs (#6676)
|
2025-08-07 12:51:47 -04:00 |
|
drafter.py
|
[TRTLLM-8136][feat] Dynamic draft length in spec decode (stage 1). (#8194)
|
2025-11-18 11:13:39 -05:00 |
|
drafting_loops.py
|
[https://nvbugs/5772414][fix] Fix draft token tree depth=1 corner case (#10385)
|
2026-01-05 17:20:14 +01:00 |
|
eagle3.py
|
[TRTLLM-10321][feat] Support different KV cache layout for one-model spec dec (#10502)
|
2026-02-10 05:16:02 +08:00 |
|
interface.py
|
[https://nvbugs/5853720][fix] Disable cutedsl argmax kernel to fix perf regression (#11403)
|
2026-02-10 18:10:38 +08:00 |
|
model_drafter.py
|
[TRTLLM-9962][feat] Some optimizations for two-model spec dec (#10208)
|
2025-12-28 12:52:04 +08:00 |
|
mtp.py
|
[TRTLLM-10321][feat] Support different KV cache layout for one-model spec dec (#10502)
|
2026-02-10 05:16:02 +08:00 |
|
ngram.py
|
[TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586)
|
2025-11-25 09:40:55 -05:00 |
|
one_model_sampler.py
|
[None][feat] Speculative One Model: FlashInfer sampling (#10284)
|
2026-01-20 12:56:43 -05:00 |
|
save_hidden_state.py
|
[TRTLLM-8160][feat] Add max_total_draft_tokens (#8366)
|
2025-10-21 11:11:04 -04:00 |
|
spec_tree_manager.py
|
[TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586)
|
2025-11-25 09:40:55 -05:00 |
|
speculation_gate.py
|
[TRTLLM-7412][feat] Turn off spec decode when the rolling average acceptance length drops below threshold. (#7283)
|
2025-10-13 15:51:14 -07:00 |
|
utils.py
|
[TRTLLM-10321][feat] Support different KV cache layout for one-model spec dec (#10502)
|
2026-02-10 05:16:02 +08:00 |
