github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
27ab5240965fc9cc0ec74ff48356abcbf098bd74
ColossalAI/colossalai/auto_parallel/tensor_shard/solver
History
YuliangLiu0306 845ff4a47a [autoparallel] resnet block runtime apply (#1709) 
* [autoparallel] resnet block runtime apply

* seperate buffer and parameter in MemoryCost

* polish code

* add comments and todos

* fix test issue
2022-10-17 13:37:38 +08:00
..
__init__.py
[autoparallel] refactored the autoparallel module for organization (#1706)
2022-10-14 13:27:00 +08:00
cost_graph.py
[autoparallel] resnet block runtime apply (#1709)
2022-10-17 13:37:38 +08:00
graph_analysis.py
[autoparallel] refactored the autoparallel module for organization (#1706)
2022-10-14 13:27:00 +08:00
options.py
[autoparallel] refactored the autoparallel module for organization (#1706)
2022-10-14 13:27:00 +08:00
solver.py
[autoparallel] resnet block runtime apply (#1709)
2022-10-17 13:37:38 +08:00
strategies_constructor.py
[autoparallel] refactored the autoparallel module for organization (#1706)
2022-10-14 13:27:00 +08:00