mirror of https://github.com/hpcaitech/ColossalAI
8accecd55b
* [legacy] move engine to legacy * [example] fix seq parallel example * [example] fix seq parallel example * [test] test gemini pluging hang * [test] test gemini pluging hang * [test] test gemini pluging hang * [test] test gemini pluging hang * [test] test gemini pluging hang * [example] update seq parallel requirements |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _base_gradient_handler.py | ||
| _data_parallel_gradient_handler.py | ||
| _moe_gradient_handler.py | ||
| _pipeline_parallel_gradient_handler.py | ||
| _sequence_parallel_gradient_handler.py | ||
| _zero_gradient_handler.py | ||
| utils.py | ||