TensorRT-LLMs/tests/unittest/_torch/modeling
Perkz Zheng 71ccc07d2b
[None][feat] update trtllm-gen to support groupsTokensHeadsQ (#10261)
Signed-off-by: Perkz Zheng <67892460+PerkzZheng@users.noreply.github.com>
Signed-off-by: Bo Li <22713281+bobboli@users.noreply.github.com>
Co-authored-by: Bo Li <22713281+bobboli@users.noreply.github.com>
2026-01-15 02:24:25 -05:00
..
test_modeling_bert.py feat: no-cache attention in PyTorch workflow (#3085) 2025-04-05 01:54:32 +08:00
test_modeling_clip.py feat: add Pytorch support of Vision Encoder for multimodal models (#3791) 2025-05-03 05:13:47 +08:00
test_modeling_exaone4.py [https://nvbugs/5569713][fix] Disable fp8 deep gemm for EXAONE-4.0-32B-FP8 (#8429) 2025-11-20 12:43:13 -05:00
test_modeling_exaone_moe.py [TRTLLM-10195][feat] K-EXAONE support (#10355) 2026-01-12 00:29:51 +09:00
test_modeling_gemma3.py [None][chore] Update Gemma3 closeness check to mitigate flakiness (#6591) 2025-08-04 10:10:58 -04:00
test_modeling_gpt_oss.py [None][fix] Fix dummy load format for key models. (#7993) 2025-10-14 11:18:39 +08:00
test_modeling_llama_min_latency.py [TRTLLM-8310][feat] Add Qwen3-VL-MoE (#9689) 2025-12-15 20:05:20 -08:00
test_modeling_llama.py [None][feat] update trtllm-gen to support groupsTokensHeadsQ (#10261) 2026-01-15 02:24:25 -05:00
test_modeling_llava_next.py [TRTLLM-6928][fix] Refactor multimodal unittest (#8453) 2025-11-03 06:01:07 -08:00
test_modeling_mistral.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
test_modeling_mixtral.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
test_modeling_mllama.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
test_modeling_multimodal.py [TRTLLM-8310][feat] Add Qwen3-VL-MoE (#9689) 2025-12-15 20:05:20 -08:00
test_modeling_nemotron_h.py [https://nvbugs/5721644][fix] Update tests for nemotron_h (#9993) 2025-12-18 12:38:02 +08:00
test_modeling_nemotron_nano_v2_vl.py [TRTLLM-8119][feat] Update doc/tests/chat_template for nano-v2-vlm (#8840) 2025-11-11 07:48:23 -08:00
test_modeling_nemotron_nas.py [None][chore] Upgrade transformers to 4.56.0 (#7523) 2025-09-22 22:20:16 +08:00
test_modeling_nemotron.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
test_modeling_out_of_tree.py [TRTLLM-9191][feat] support out-of-tree models in trtllm-serve (#9269) 2025-11-21 04:23:47 -08:00
test_modeling_phi3.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
test_modeling_phi4mm.py [TRTLLM-9264][fix] Add accuracy/unit tests/doc for phi4mm (#9246) 2025-11-26 11:12:35 +08:00
test_modeling_pixtral.py [None][chore] update torch_dtype -> dtype in 'transformers' (#8263) 2025-10-15 17:09:30 +09:00
test_modeling_qwen2_5vl.py [TRTLLM-8310][feat] Add Qwen3-VL-MoE (#9689) 2025-12-15 20:05:20 -08:00
test_modeling_qwen3vl_moe.py [TRTLLM-8310][feat] Add Qwen3-VL-MoE (#9689) 2025-12-15 20:05:20 -08:00
test_modeling_qwen3vl.py [None][fix] Enable AttentionDP on Qwen3-VL and fix test (#10435) 2026-01-10 00:13:26 +09:00
test_modeling_qwen_moe.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
test_modeling_qwen.py [TRTLLM-8521][chore] remove circular dependency between model engine and cuda graph runner (#7572) 2025-11-11 10:13:45 -08:00
test_modeling_siglip.py [TRTLLM-8310][feat] Add Qwen3-VL-MoE (#9689) 2025-12-15 20:05:20 -08:00
test_modeling_starcoder2.py [TRTLLM-7967][chore] Add more tests (#9415) 2025-12-08 11:57:32 -08:00
test_modeling_vila.py [None][chore] use cached vila model (#8788) 2025-10-30 20:26:45 -07:00