github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/fx/passes/algorithms
History
Boyuan Yao 31fffd3fc5
[fx] fix wrong variable name in solver rotor (#1502) 
* [fx] fix wrong variable name in solver rotor

* [fx] fix wrong variable name in solver rotor

* code modification
 		2022-08-26 15:47:08 +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 wrong variable name in solver rotor (#1502) 2022-08-26 15:47:08 +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