2 Commits (573af8418406a319e91be07f58fca798a6e72dbd)

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