TensorRT-LLMs/tests/unittest/_torch/modules
hlu1 320195dc0d
[Architecture] Refactor FusedMoE (#4790)
Signed-off-by: Hao Lu <14827759+hlu1@users.noreply.github.com@users.noreply.github.com>
Co-authored-by: Hao Lu <14827759+hlu1@users.noreply.github.com@users.noreply.github.com>
2025-06-03 14:02:19 +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 [Architecture] Refactor FusedMoE (#4790) 2025-06-03 14:02:19 +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