mirror of https://github.com/hpcaitech/ColossalAI
8ea50d999e
* [autockpt] make it work. * [autockpt] linearize / merge shape-consistency nodes. * [autockpt] considering parameter and optimizer weights. * [hotfix] pass a parameter. |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| build_c_ext.py | ||
| ckpt_solver_base.py | ||
| ckpt_solver_chen.py | ||
| ckpt_solver_rotor.c | ||
| ckpt_solver_rotor.py | ||
| operation.py | ||