TensorRT-LLMs/tests/unittest/_torch/speculative
Anish Shanbhag e308eb50f4
[TRTLLM-10803][fix] Fix mocking of HuggingFace downloads in with_mocked_hf_download (#11200)
Signed-off-by: Anish Shanbhag <ashanbhag@nvidia.com>
2026-02-02 21:58:15 -08:00
..
test_draft_len_schedule.py [https://nvbugs/5680911][fix] Remove @cache decorator to enhance CI stability for unit tests using single process mode (#10730) 2026-02-02 16:26:46 +08:00
test_draft_target.py [None][feat] Auto download speculative models from HF for pytorch backend, add speculative_model field alias (#10099) 2026-01-14 21:06:07 -08:00
test_draft_token_prepare_for_generation.py [TRTC-122][feat] Eagle3 Specdec UX improvements (#10124) 2026-01-22 07:24:11 -08:00
test_draft_token_tree_sampling.py [TRTC-122][feat] Eagle3 Specdec UX improvements (#10124) 2026-01-22 07:24:11 -08:00
test_draft_token_tree_verification.py [TRTC-122][feat] Eagle3 Specdec UX improvements (#10124) 2026-01-22 07:24:11 -08:00
test_dynamic_spec_decode.py [https://nvbugs/5680911][fix] Remove @cache decorator to enhance CI stability for unit tests using single process mode (#10730) 2026-02-02 16:26:46 +08:00
test_eagle3.py [TRTLLM-10803][fix] Fix mocking of HuggingFace downloads in with_mocked_hf_download (#11200) 2026-02-02 21:58:15 -08:00
test_kv_cache_reuse.py [TRTC-122][feat] Eagle3 Specdec UX improvements (#10124) 2026-01-22 07:24:11 -08: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 [https://nvbugs/5680911][fix] Remove @cache decorator to enhance CI stability for unit tests using single process mode (#10730) 2026-02-02 16:26:46 +08: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