ColossalAI

Commit Graph

Author	SHA1	Message	Date
YuliangLiu0306	ffcdbf0f65	[autoparallel]integrate auto parallel feature with new tracer (#3408 ) * [autoparallel] integrate new analyzer in module level * unify the profiling method * polish * fix no codegen bug * fix pass bug * fix liveness test * polish	2023-04-04 17:40:45 +08:00
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

Author

SHA1

Message

Date

YuliangLiu0306

ffcdbf0f65

[autoparallel]integrate auto parallel feature with new tracer (#3408 )

* [autoparallel] integrate new analyzer in module level

* unify the profiling method

* polish

* fix no codegen bug

* fix pass bug

* fix liveness test

* polish

2023-04-04 17:40:45 +08:00

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

3 Commits (839847b7d78bce6af5dfe58d27b5ce2c74a3619b)