TensorRT-LLMs/tests/unittest/_torch/modules
Yuxian Qiu 6b3242654e
fix: Fix broken vanilla moe since FusedMoE refactor. (#4897)
Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com>
2025-06-05 03:56:41 +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 fix: Fix broken vanilla moe since FusedMoE refactor. (#4897) 2025-06-05 03:56:41 +08:00
test_moe_host_sharer.py [Architecture] Refactor FusedMoE (#4790) 2025-06-03 14:02:19 +08:00
test_moe_load_balancer.py [Architecture] Refactor FusedMoE (#4790) 2025-06-03 14:02:19 +08:00
test_moe_routing.py chore: reorganize some unit tests of PyTorch (#3780) 2025-04-23 11:19:10 -07:00