mirror of https://github.com/hpcaitech/ColossalAI
bb0a668fee
* fix: simplify merge_batch * fix: use return_outputs=False to eliminate extra memory consumption * feat: add return_outputs warning * style: remove `return_outputs=False` as it is the default value |
||
---|---|---|
.. | ||
moe_utils.py | ||
test_grad_handler.py | ||
test_kernel.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_router.py | ||
test_moe_zero_fwd_bwd.py | ||
test_moe_zero_optim.py |