github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-04 08:35:26 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,618 Commits 59 Branches 51 Tags
1aaeb596c63752071cbbaa2477a7f2406901b70b
Commit Graph

2 Commits

Author SHA1 Message Date
Super Daniel
3ccf58aa76 [autockpt] make it work. (#2257) 2023-01-02 23:37:45 +08:00
Boyuan Yao
ab38aebace [autoparallel] Hook all meta information on ResNet nodes for auto activation checkpoint (#2248) 
* [autoparallel] hook node meta on graph nodes for checkpoint solver

* [autoparallel] polish code

* [autoparallel] restore some node handlers

* colossalai/auto_parallel/passes/meta_info_prop.py

* [autoparallel] remove some unused import

* [autoparallel] hook bwd_mem_out
 2023-01-02 16:25:18 +08:00