github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/auto_parallel/solver
History
YuliangLiu0306 413c053453
[autoparallel] add cost graph class (#1481) 
* [autoparallel] add cost graph class

* polish code
 		2022-08-25 17:19:59 +08:00
..
__init__.py [autoparallel] integrate auto parallel with torch fx (#1479) 2022-08-23 14:23:08 +08:00
conv_handler.py [autoparallel] added namespace constraints (#1490) 2022-08-24 15:44:07 +08:00
cost_graph.py [autoparallel] add cost graph class (#1481) 2022-08-25 17:19:59 +08:00
dot_handler.py [autoparallel] added namespace constraints (#1490) 2022-08-24 15:44:07 +08:00
operator_handler.py [autoparallel] add cost graph class (#1481) 2022-08-25 17:19:59 +08:00
sharding_strategy.py [autoparallel] add cost graph class (#1481) 2022-08-25 17:19:59 +08:00