mirror of https://github.com/hpcaitech/ColossalAI
![]() * 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 |
||
---|---|---|
.. | ||
test_pipeline_utils | ||
test_schedule | ||
test_p2p_communication.py | ||
test_stage_manager.py |