TensorRT-LLMs/tests/unittest/_torch/sampler
mpikulski 7c8ba71b49
[TRTLLM-8832][feat] fully async _select_generated_logits with tests (#8628)
Signed-off-by: ixlmar <206748156+ixlmar@users.noreply.github.com>
2025-10-27 16:15:32 +01:00
..
test_beam_search.py [None][ci] move unittests to sub-directories (#6635) 2025-08-20 05:42:22 -04: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