TensorRT-LLMs/tests/unittest/_torch/sampler
Michal Guzek 8e9409ce04
[https://nvbugs/5628204][fix] Stop token IDs - fast path optimization for single stop token IDs only (#9014)
Signed-off-by: Michal Guzek <mguzek@nvidia.com>
Signed-off-by: Michal Guzek <moraxu@users.noreply.github.com>
2025-11-13 14:17:20 +01:00
..
test_beam_search.py [TRTLLM-8650][fix] beam search request validation (#8433) 2025-10-21 10:50:27 +02:00
test_best_of_n.py [None][ci] move unittests to sub-directories (#6635) 2025-08-20 05:42:22 -04:00
test_return_logits.py [TRTLLM-7155][feat] Unify sampler handle logits implementation. (#6867) 2025-08-22 08:09:30 +02:00
test_trtllm_sampler.py [https://nvbugs/5628204][fix] Stop token IDs - fast path optimization for single stop token IDs only (#9014) 2025-11-13 14:17:20 +01:00