TensorRT-LLMs/tests/unittest/_torch
Yukun He b6bced83c0
[TRTLLM-7963][feat] Use CUDAGraph to improve the tuning accuracy for AutoTuner. (#9089)
Signed-off-by: Yukun He <23156053+hyukn@users.noreply.github.com>
2025-11-20 08:54:29 +08:00
..
attention [None][feat] Update the indexer topK (#9255) 2025-11-19 11:49:00 +08:00
auto_deploy [#9237][feat] enable iter stats in autodeploy (#9278) 2025-11-19 19:29:29 +01:00
compilation [TRTLLM-3105][feat] Add Piecewise CUDA Graph Support (#3804) 2025-05-09 11:04:01 +08:00
debugger Fix: fix nvbug 5356427 (#5464) 2025-06-25 22:24:26 +08:00
executor [TRTLLM-9295][fix] unflake test_overlap_scheduler.py::test_overlap_scheduler_consis… (#9146) 2025-11-14 11:36:22 +01:00
misc [TRTLLM-7963][feat] Use CUDAGraph to improve the tuning accuracy for AutoTuner. (#9089) 2025-11-20 08:54:29 +08:00
modeling [TRTLLM-8778][feat] Add tree attention support for blackwell arch (#8975) 2025-11-17 09:01:53 +08:00
models/checkpoints/hf [None][feat] Skip prefetching consolidated safetensors when appropriate (#7013) 2025-08-25 23:56:21 -04:00
modules [None][feat] Update TRTLLM MoE cubins; reduce mxfp4 weight padding requirement; tighten TMA bound (#9025) 2025-11-17 10:04:29 +08:00
multi_gpu [None][infra] Waive failed cases on main 11/05 (#8936) 2025-11-04 22:54:45 -08:00
multi_gpu_modeling [https://nvbugs/5536131][fix] Fix illegal access issue when scale is not provided in Llama3/4. (#7960) 2025-10-16 22:46:19 +08:00
multimodal [None][fix] InputProcessor config naming convention fix (#8705) 2025-11-03 22:29:21 -08:00
ray_orchestrator [None][chore] Add placement test for ray executor (#9122) 2025-11-14 23:10:59 -08:00
sampler [https://nvbugs/5661877][fix] fix test regression in TestBatchedSampling::test_samples (#9215) 2025-11-19 01:44:44 -08:00
speculative [https://nvbugs/5590408][fix] Exclude num of draft tokens from mMaxSeqLenKv (#9210) 2025-11-18 15:41:56 -05:00
thop [https://nvbugs/5456493][feat] add fp8 dense for sm120 (#9174) 2025-11-19 14:40:34 +08:00
helpers.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
pattern_watcher.py [TRTLLM-3105][feat] Add Piecewise CUDA Graph Support (#3804) 2025-05-09 11:04:01 +08:00
test_connector.py [None][feat] KV Cache Connector API (#7228) 2025-08-28 23:09:27 -04:00