mirror of https://github.com/hpcaitech/ColossalAI
fe24789eb1 | ||
---|---|---|
.. | ||
moe_utils.py | ||
test_deepseek_layer.py | ||
test_grad_handler.py | ||
test_kernel.py | ||
test_mixtral_layer.py | ||
test_moe_checkpoint.py | ||
test_moe_ep_tp.py | ||
test_moe_group.py | ||
test_moe_hybrid_zero.py | ||
test_moe_load_balance.py | ||
test_moe_zero_fwd_bwd_optim.py |