TensorRT-LLMs/tests/unittest/_torch/speculative
kris1025 a7ea544dbe
[TRTLLM-7384][feat] enable rejection sampling for CDL (#7731)
Signed-off-by: linquanh <linquanh@nvidia.com>
2025-10-12 20:38:48 +08:00
..
test_draft_target.py [TRTLLM-7457][ci] Update & cleanup unittest parallel config (#7254) 2025-08-27 00:45:58 -04:00
test_draft_token_tree_sampling.py [TRTLLM-6393][feat] add static tree sampling and verification (#7161) 2025-09-26 13:16:16 -04:00
test_draft_token_tree_verification.py [TRTLLM-6393][feat] add static tree sampling and verification (#7161) 2025-09-26 13:16:16 -04:00
test_dynamic_spec_decode.py [https://nvbugs/5548098][fix] Fix flakey unit test for dynamic spec d… (#8129) 2025-10-02 22:58:37 -07:00
test_eagle3.py [TRTLLM-7384][feat] enable rejection sampling for CDL (#7731) 2025-10-12 20:38:48 +08:00
test_kv_cache_reuse.py [TRTLLM-7457][ci] Update & cleanup unittest parallel config (#7254) 2025-08-27 00:45:58 -04:00
test_mtp.py [refactor] Simplification of Speculative decoding configs (#5639) 2025-07-10 11:37:30 -04:00
test_ngram.py [TRTLLM-7457][ci] Update & cleanup unittest parallel config (#7254) 2025-08-27 00:45:58 -04:00
test_save_state.py [None][feat] Draft: Save state first pass (#7012) 2025-10-01 18:40:55 -04:00
test_torch_rejection_sampling.py [None][feat] Add test for speculative rejection sampler (2-model) (#6542) 2025-08-13 22:09:35 -04:00
test_user_provided.py [TRTLLM-7457][ci] Update & cleanup unittest parallel config (#7254) 2025-08-27 00:45:58 -04:00