TensorRT-LLMs/tests/unittest/_torch/speculative
Zheyu Fu ab45d6a7c7 Waive dynamic spec decode unit test
Signed-off-by: Zheyu Fu <zheyuf@NVIDIA.com>
2025-12-20 01:20:06 +00:00
..
test_draft_len_schedule.py Fix thread leak for test_draft_len_schedule. Enhance stability for test_spec_gate. 2025-12-19 02:01:38 +00: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_prepare_for_generation.py [TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586) 2025-11-25 09:40:55 -05:00
test_draft_token_tree_sampling.py [TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586) 2025-11-25 09:40:55 -05:00
test_draft_token_tree_verification.py [TRTLLM-6756][feat] Add Beam Search to TorchSampler (#8509) 2025-12-01 18:48:04 +01:00
test_dynamic_spec_decode.py Waive dynamic spec decode unit test 2025-12-20 01:20:06 +00:00
test_eagle3.py [None][fix] Fix iteration stats for spec-dec (#9855) 2025-12-16 14:11:38 -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 [TRTLLM-8160][feat] Add max_total_draft_tokens (#8366) 2025-10-21 11:11:04 -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_spec_gate.py Fix thread leak for test_draft_len_schedule. Enhance stability for test_spec_gate. 2025-12-19 02:01:38 +00:00
test_torch_rejection_sampling.py [None][fix] restore list[list[list[int]]] in add_token (#8502) 2025-10-20 22:34:57 -04:00
test_user_provided.py [TRTLLM-7457][ci] Update & cleanup unittest parallel config (#7254) 2025-08-27 00:45:58 -04:00