github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/auto_parallel/tensor_shard
History
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
..
node_handler [autoparallel]integrate auto parallel feature with new tracer (#3408) 2023-04-04 17:40:45 +08:00
solver [autoparallel]integrate auto parallel feature with new tracer (#3408) 2023-04-04 17:40:45 +08:00
utils [autoparallel] find repeat blocks (#2854) 2023-02-23 17:28:19 +08:00
__init__.py
…
constants.py
…
initialize.py [autoparallel]integrate auto parallel feature with new tracer (#3408) 2023-04-04 17:40:45 +08:00
options.py [autoparallel] add shard option (#2696) 2023-02-15 13:48:28 +08:00
sharding_strategy.py [autoparallel] memory estimation for shape consistency (#2144) 2022-12-21 10:39:37 +08:00