digger yu
|
7f8203af69
|
fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)
|
2023-05-24 09:01:50 +08:00 |
YuliangLiu0306
|
314d8c497f
|
[autoparallel] refactor the runtime apply pass and add docstring to passes (#1757)
* [autoparallel] refactor the runtime apply pass and add doc string to passes
* fix unit test
* polish
|
2022-10-25 14:32:22 +08:00 |
YuliangLiu0306
|
d2fc067231
|
[autoparallel] fix param hook issue in transform pass (#1755)
|
2022-10-24 13:13:38 +08:00 |
YuliangLiu0306
|
980ed21723
|
[autoparallel] shard param and buffer as expected (#1753)
* [autoparallel] shard param and buffer as expected
* fix unit test issue
|
2022-10-21 15:45:13 +08:00 |
YuliangLiu0306
|
a4ce180e85
|
[autoparallel] add sequential order to communication actions (#1735)
|
2022-10-20 18:48:18 +08:00 |
Frank Lee
|
993b8875b6
|
[autoparallel] handled illegal sharding strategy in shape consistency (#1744)
* [autoparallel] handled illegal sharding strategy in shape consistency
* polish code
|
2022-10-20 12:06:25 +08:00 |
YuliangLiu0306
|
51b89d2202
|
[autoparallel] runtime_backward_apply (#1720)
|
2022-10-18 10:44:58 +08:00 |
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 |
YuliangLiu0306
|
451cd72dea
|
[autoparallel] adapt runtime passes (#1703)
* [autoparallel] adapt runtime passes v2
* polish code
|
2022-10-14 10:14:07 +08:00 |
YuliangLiu0306
|
7d1bb71d5d
|
[fx] PoC of runtime shape consistency application (#1607)
* [fx] PoC of runtime shape consistency application
* polish code
|
2022-09-20 14:00:04 +08:00 |