TensorRT-LLMs/tests/unittest/utils
mpikulski c5f52ab304
[TRTLLM-8376][feat] top-p optimization (removes redundant softmax) (#9411)
Signed-off-by: ixlmar <206748156+ixlmar@users.noreply.github.com>
2025-11-25 18:46:48 +01:00
..
__init__.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
cpp_paths.py chore: fix llm_root when LLM_ROOT is not set (#4741) 2025-05-29 19:44:34 -07:00
llm_data.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
runtime_defaults.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_medusa_utils.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_prebuilt_whl_cpp_extensions.py [https://nvbugs/5617275][fix] Extract py files from prebuilt wheel for editable installs (#8738) 2025-10-30 21:40:22 -07:00
test_util.py [TRTLLM-8832][feat] fully async _select_generated_logits with tests (#8628) 2025-10-27 16:15:32 +01:00
torch_ref.py [TRTLLM-4921][feat] Enable chunked prefill for Nemotron-H (#6334) 2025-08-22 12:15:20 -04:00
util.py [TRTLLM-8376][feat] top-p optimization (removes redundant softmax) (#9411) 2025-11-25 18:46:48 +01:00