TensorRT-LLMs/tests/unittest/_torch/sampler
Stefan Niebler 19ca7b15c7
[https://nvbugs/5593199][test] Enhance beam search tests deterministic dummy model (#8625)
Signed-off-by: Stefan Niebler <82932102+stnie@users.noreply.github.com>
2025-10-29 06:12:22 +01:00
..
test_beam_search.py [https://nvbugs/5593199][test] Enhance beam search tests deterministic dummy model (#8625) 2025-10-29 06:12:22 +01: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-8551][feat] add cache_salt in LLM.generate and refactor test_return_logits.py (#8317) 2025-10-15 02:53:57 -07:00
test_torch_multi_arange.py [TRTLLM-8832][feat] fully async _select_generated_logits with tests (#8628) 2025-10-27 16:15:32 +01:00
test_torch_sampler.py [TRTLLM-8832][feat] fully async _select_generated_logits with tests (#8628) 2025-10-27 16:15:32 +01:00
test_trtllm_sampler.py [TRTLLM-8414][chore] BREAKING CHANGE: refine sampling strategy selection (#8132) 2025-10-08 15:46:50 +02:00