TensorRT-LLMs/tests/unittest/_torch/sampler
mpikulski 899fda9e47
[TRTLLM-9490][feat] use FlashInfer's top_k_sampling_from_probs (#9457)
Signed-off-by: ixlmar <206748156+ixlmar@users.noreply.github.com>
2025-11-25 18:53:53 +01:00
..
test_beam_search.py [TRTLLM-8650][fix] beam search request validation (#8433) (#9228) 2025-11-21 04:08:45 -08: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-9302][chore] Move build config from BaseLlmArgs to TrtLlmArgs (#9249) 2025-11-24 10:54:41 +08: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-9490][feat] use FlashInfer's top_k_sampling_from_probs (#9457) 2025-11-25 18:53:53 +01:00
test_trtllm_sampler.py [TRTLLM-8414][chore] BREAKING CHANGE: refine sampling strategy selection (#8132) 2025-10-08 15:46:50 +02:00