TensorRT-LLMs/tests/unittest/_torch/sampler
mpikulski 46dd9886bb
[https://nvbugs/5661877][fix] fix test regression in TestBatchedSampling::test_samples (#9215)
Signed-off-by: ixlmar <206748156+ixlmar@users.noreply.github.com>
2025-11-19 01:44:44 -08: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-8831][feat] Enable early exit with overlap scheduler (#8587) 2025-11-17 18:07:13 +01: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 [https://nvbugs/5661877][fix] fix test regression in TestBatchedSampling::test_samples (#9215) 2025-11-19 01:44:44 -08:00
test_trtllm_sampler.py [TRTLLM-8414][chore] BREAKING CHANGE: refine sampling strategy selection (#8132) 2025-10-08 15:46:50 +02:00