mirror of https://github.com/hpcaitech/ColossalAI
56159049e8
* [fx] modify solver linearize and add test * [fx] add torch11 test of linearize but skip it * [fx] remove some unused imports |
||
---|---|---|
.. | ||
__init__.py | ||
ckpt_solver_chen.py | ||
ckpt_solver_rotor.py | ||
linearize.py | ||
utils.py |