mirror of https://github.com/hpcaitech/ColossalAI
d967779a32
* [fx] tuned the meta info and rotor solver. * [fx] remove import. * [fx] remove import. * [fx] remove import. * [fx] tune the meta calculations. * [fx] polish comments. * [fx] remove assertions. * [fx] modify test cases. * [fx] modify test cases. * [fx] optimize import. * [fx |
||
---|---|---|
.. | ||
__init__.py | ||
ckpt_solver_chen.py | ||
ckpt_solver_pofo.py | ||
ckpt_solver_rotor.py | ||
linearize.py | ||
operation.py |