Boyuan Yao
|
24246f7aa5
|
[autoparallel] Attach input, buffer and output tensor to MetaInfo class (#2162)
* [fx] metainfo class for auto parallel
* [fx] add unit test for linear metainfo
* [fx] fix bwd param for linear
* [fx] modify unit test
* [fx] modify unit test
* [fx] modify import
* [fx] modify import
* [fx] modify import
* [fx] move meta profiler to auto parallel
* [fx] add conv metainfo class
* [fx] restore profiler
* [fx] restore meta profiler
* [autoparallel] modify unit test
* [fx] modify unit test
* [autoparallel] add batchnorm metainfo class
* [autoparallel] fix batchnorm unit test function declaration
* [fx] restore profiler
* [fx] add relu metainfo class
* [fx] restore profiler
* [autoparallel] modify metainfo input
* [autoparallel] add pooling metainfo
* [autoparallel] add F.linear metainfo generator
* [autoparallel] add binary elementwise metainfo
* [fx] recover profiler
* [autoparallel] fix forward memory calculation
* [autoparallel] modify constants.py
* [autoparallel] remove redundant print
* [autoparallel] add F.conv metainfo
* [autoparallel] linear fix
* [autoparallel] memory estimation for communication actions
* [autoparallel] fix docstring
* [autoparallel] fix variables name
* [autoparallel] attach tensor to metainfo class
* [autoparallel] fix dangerous try except
* [autoparallel] attach memory cost to shape consistency node
* [autoparallel] attach shape consistency node's metainfo to the node
* [autoparallel] remove todo in shape consistency memory estimation
* [autoparallel] fix the annotation
|
2022-12-28 13:37:40 +08:00 |
|
Boyuan Yao
|
d0bc5a1b34
|
[autoparallel] new metainfoprop based on metainfo class (#2179)
* [autoparallel] new metainfoprop to combine SPMD solver and checkpoint solver
* [autoparallel] new metainfoprop to combine SPMD solver and checkpoint solver
* [autoparallel] modify placeholder handler
* [autoparallel] modify metainfoprop
* [autoparallel] fix function typo
* [autoparallel] fix placeholder handler
|
2022-12-28 13:35:08 +08:00 |
|
YuliangLiu0306
|
78509124d3
|
[autoparallel] update getitem handler (#2207)
|
2022-12-27 19:58:32 +08:00 |
|
YuliangLiu0306
|
4851f2d607
|
[autoparallel] update_getattr_handler (#2193)
|
2022-12-26 21:57:39 +08:00 |
|
YuliangLiu0306
|
550f8f8905
|
[autoparallel] integrate_gpt_related_tests (#2134)
* [autoparallel] integrate_gpt_related_tests
* polish code
* polish code
* add GPT2Model into runtime test
|
2022-12-23 12:36:59 +08:00 |
|
YuliangLiu0306
|
a3c6924deb
|
[autoparallel] process size nodes in runtime pass (#2130)
* [autoparallel] process size nodes in runtime pass
* polish code
|
2022-12-14 16:10:50 +08:00 |
|
YuliangLiu0306
|
cd0af9f7f6
|
[autoparallel] gpt2lp runtimee test (#2113)
|
2022-12-12 18:06:40 +08:00 |
|
YuliangLiu0306
|
e4293e5077
|
[hotfix] update test for latest version (#2060)
|
2022-12-02 18:12:30 +08:00 |
|
YuliangLiu0306
|
0dbcd4a6f5
|
[autoparallel] add split handler (#2032)
* [autoparallel] add split handler
* add numerical test and runtime passes
|
2022-11-29 11:03:51 +08:00 |
|
YuliangLiu0306
|
81330b0352
|
[autoparallel] add experimental permute handler (#2029)
|
2022-11-27 20:26:52 +08:00 |
|
YuliangLiu0306
|
ea0f6b8df9
|
[autoparallel] add runtime pass and numerical test for view handler (#2018)
|
2022-11-25 15:50:16 +08:00 |
|
YuliangLiu0306
|
36c0f3ea5b
|
[autoparallel] remove redundancy comm node (#1893)
|
2022-11-15 10:53:41 +08:00 |
|
YuliangLiu0306
|
1b494ad73c
|
[autoparallel] fix linear logical convert issue (#1857)
|
2022-11-10 17:19:22 +08:00 |
|
YuliangLiu0306
|
f6032ddb17
|
[autoparallel] fix bias addition module (#1800)
|
2022-11-08 16:21:25 +08:00 |
|
YuliangLiu0306
|
b4cc59b61e
|
[autoparallel] add numerical test for node strategies (#1760)
* [autoparallel] add numerical test for node strategies
* polish code
* polish code
|
2022-10-27 10:42:54 +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 |