TensorRT-LLMs/tests/unittest/utils
shuyixiong 70e4d72ffa
[TRTLLM-8511][feat] Add update_weights and sleep_wakeup support for rl integration (#8302)
Signed-off-by: shuyix <219646547+shuyixiong@users.noreply.github.com>
Co-authored-by: Liwei Ma <liweim@nvidia.com>
Co-authored-by: Jonas Yang CN <joyang@nvidia.com>
2025-11-04 10:19:24 -08: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-8511][feat] Add update_weights and sleep_wakeup support for rl integration (#8302) 2025-11-04 10:19:24 -08:00