mirror of https://github.com/hpcaitech/ColossalAI
f7f2248771
* remove forced FP32 modules * correct no_shard-contexts' positions |
||
---|---|---|
.. | ||
test_grad_handler.py | ||
test_kernel.py | ||
test_moe_group.py | ||
test_moe_zero_init.py | ||
test_moe_zero_model.py | ||
test_moe_zero_optim.py |