TensorRT-LLMs/tests/unittest/_torch/modules
Yuxian Qiu 5700a4ffcd
feat: Add vanilla MOE. (#4682)
Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com>
2025-05-28 10:44:14 +08:00
..
tests_lora_modules added loraOp into lora layer + test for mlp and comparison to lora plugin (#3455) 2025-04-17 12:48:27 +08:00
test_fused_moe.py feat: Add vanilla MOE. (#4682) 2025-05-28 10:44:14 +08:00
test_moe_host_sharer.py feat: large-scale EP(part 2: MoE Load Balancer - core utilities) (#4384) 2025-05-20 17:53:48 +08:00
test_moe_load_balancer.py feat: large-scale EP(part 2: MoE Load Balancer - core utilities) (#4384) 2025-05-20 17:53:48 +08:00
test_moe_routing.py chore: reorganize some unit tests of PyTorch (#3780) 2025-04-23 11:19:10 -07:00