TensorRT-LLMs/tests/unittest/_torch
Yechan Kim f48968b6cc
[TRTLLM-6928][fix] Refactor multimodal unittest (#8453)
Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com>
2025-11-03 06:01:07 -08:00
..
attention [TRTLLM-8541][feat] Add trtllm-gen sparse MLA kernels to support per-Tensor FP8 KV Cache (#8692) 2025-10-31 14:38:31 -07:00
auto_deploy [https://nvbugs/5606166][fix] AutoDeploy: use tuples for cudagraph shape lookup (#8658) 2025-10-28 10:52:43 -07: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-8836][chore] Create ModelEngine from LlmArgs (#8600) 2025-11-01 05:26:06 -07:00
misc [None][feat] Autotuner can iterate through all tactics for test purposes (#8663) 2025-10-30 13:11:25 +01:00
modeling [TRTLLM-6928][fix] Refactor multimodal unittest (#8453) 2025-11-03 06:01:07 -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] Autotuner can iterate through all tactics for test purposes (#8663) 2025-10-30 13:11:25 +01:00
multi_gpu [TRTLLM-7318][feat] MnnvlThroughput AlltoAll implementation. (#7499) 2025-10-27 13:23:06 -04: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 [https://nvbugs/5625380][chore] Remove multimodal related fields from decoder llm input (#8846) 2025-11-02 17:44:08 -08:00
ray_orchestrator [TRTLLM-8513][feat] Add back worker extension (#8482) 2025-10-24 20:30:28 -04:00
sampler [https://nvbugs/5593199][test] Enhance beam search tests deterministic dummy model (#8625) 2025-10-29 06:12:22 +01:00
speculative [https://nvbugs/5534574][fix] disable spec decoding forever once the request spec decoding is disabled (#8446) 2025-10-29 19:28:43 +08:00
thop [https://nvbugs/5474119][fix] Re-enable test (#8809) 2025-10-31 10:17:58 -07:00
helpers.py [TRTLLM-8836][chore] Create ModelEngine from LlmArgs (#8600) 2025-11-01 05:26:06 -07: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