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 [fx] Add activation checkpoint solver rotor (#1496) 2022-08-26 10:34:21 +08:00
ckpt_solver_chen.py [fx] add profiler for fx nodes. (#1480) 2022-08-24 16:22:44 +08:00
ckpt_solver_rotor.py [fx] fix the discretize bug (#1506) 2022-08-26 17:15:52 +08:00
linearize.py [fx] Add activation checkpoint solver rotor (#1496) 2022-08-26 10:34:21 +08:00
utils.py [fx] fix wrong variable name in solver rotor (#1502) 2022-08-26 15:47:08 +08:00