ColossalAI/colossalai/fx/passes/algorithms
Boyuan Yao ac3a453a50
[fx] fix the discretize bug (#1506)
* [fx] fix wrong variable name in solver rotor

* [fx] fix wrong variable name in solver rotor

* code modification

* [fx] fix the discretize bug
2022-08-26 17:15:52 +08:00
..
__init__.py
ckpt_solver_chen.py
ckpt_solver_rotor.py [fx] fix the discretize bug (#1506) 2022-08-26 17:15:52 +08:00
linearize.py
utils.py