github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-07 16:08:22 +00:00
Code Issues Packages Projects Releases Wiki Activity
7077d38d5a
ColossalAI/tests/test_moe
History
hxwang 7077d38d5a [moe] finalize test (no pp)
2024-08-01 10:06:59 +08:00
..
modelling [moe] finalize test (no pp) 2024-08-01 10:06:59 +08:00
moe_utils.py [moe] clean legacy code 2024-08-01 10:06:59 +08:00
test_deepseek_layer.py [shardformer] DeepseekMoE support () 2024-07-05 16:13:58 +08:00
test_kernel.py [moe] clean legacy code 2024-08-01 10:06:59 +08:00
test_moe_checkpoint.py [moe] init moe plugin comm setting with sp 2024-08-01 10:06:59 +08:00
test_moe_ep_tp.py [moe] test deepseek 2024-08-01 10:06:59 +08:00
test_moe_ep_zero.py [moe] test deepseek 2024-08-01 10:06:59 +08:00