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_3d_plugin.py | ||
test_dp_plugin_base.py | ||
test_gemini_plugin.py | ||
test_low_level_zero_plugin.py | ||
test_torch_ddp_plugin.py | ||
test_torch_fsdp_plugin.py |