18 Commits (feature/inference-refactor)

Author SHA1 Message Date
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752) 1 year ago
digger yu 7f8203af69
fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808) 2 years ago
digger-yu b9a8dff7e5
[doc] Fix typo under colossalai and doc(#3618) 2 years ago
Super Daniel 1e88811c7a
[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2 years ago
Boyuan Yao 5d0fdb9cb4
[fx] fix offload codegen test (#1648) 2 years ago
Boyuan Yao f921733621
[autoparallel] Add pofo sequence annotation (#1637) 2 years ago
Boyuan Yao d6b01feb66
[fx] Modify offload codegen (#1618) 2 years ago
Boyuan Yao a7cda6f57d
[fx] Add offload codegen (#1598) 2 years ago
Boyuan Yao f3687e4ee2
[fx] Add nested checkpoint in activation checkpoint codegen (#1585) 2 years ago
Boyuan Yao 56159049e8
[fx] Modify solver linearize and add corresponding test (#1531) 2 years ago
Boyuan Yao 4acc58ee20
[fx] Fix activation codegen dealing with checkpointing first op (#1510) 2 years ago
Boyuan Yao 1f2e547f7a
[fx] Fix ckpt functions' definitions in forward (#1476) 2 years ago
Boyuan Yao 092b9c8f49
[fx] Add use_reentrant=False to checkpoint in codegen (#1463) 2 years ago
Jiarui Fang b1553fdf96
[NFC] global vars should be upper case (#1456) 2 years ago
Boyuan Yao 5774fe0270
[fx] Use colossalai checkpoint and add offload recognition in codegen (#1439) 2 years ago
Frank Lee 89e60d1505
[fx] fixed indentation error in checkpointing codegen (#1385) 2 years ago
Frank Lee cd063ac37f
[fx] added activation checkpoint codegen support for torch < 1.12 (#1359) 2 years ago
Frank Lee 644582eee9
[fx] added activation checkpoint codegen (#1355) 2 years ago