Boyuan Yao
|
49ccf8b5f8
|
[fx] Improve linearize and rotor solver (#1586)
* [fx] add nested activation_checkpoint codegen
* undo algorithms commits
* solver
* undo some commits
* [fx] torch11 add nested activation checkpoint codegen
* remove some imports
* [fx] add some comments in activation codegen
* [fx] codegen instance error fix
* [fx] imporve linearize and rotor solver
* [fx] some comments and format modification
|
2022-09-13 14:50:04 +08:00 |
|
Boyuan Yao
|
46c6cc79a9
|
[fx] Add common node in model linearize (#1542)
* [fx] Add common node into linearize
* [fx] Add common node to solver
|
2022-09-05 18:35:05 +08:00 |
|
Boyuan Yao
|
56159049e8
|
[fx] Modify solver linearize and add corresponding test (#1531)
* [fx] modify solver linearize and add test
* [fx] add torch11 test of linearize but skip it
* [fx] remove some unused imports
|
2022-09-02 10:24:41 +08:00 |
|
Boyuan Yao
|
de1e716dc4
|
[fx] Add activation checkpoint solver rotor (#1496)
* [fx] fix defining ckpt functions inside forward
* [fx] Modify activation checkpoint codegen and add ColoGraphModule
* [fx] some modification
* some modifications
* some modifications
* some modifications
* some modifications
* some code modifications
* [automatic_parallel] ckpt solver rotor
* [fx] add ckpt_solver_rotor
* [fx] modification
* code refactor
* code refactor
|
2022-08-26 10:34:21 +08:00 |