Frank Lee
|
c4b1b65931
|
[test] fixed tests failed due to dtensor change (#4082)
* [test] fixed tests failed due to dtensor change
* polish code
|
2023-07-04 16:05:01 +08:00 |
Frank Lee
|
611971248c
|
[device] support init device mesh from process group (#3990)
|
2023-07-04 16:05:01 +08:00 |
github-actions[bot]
|
a52f62082d
|
[format] applied code formatting on changed files in pull request 4021 (#4022)
Co-authored-by: github-actions <github-actions@github.com>
|
2023-06-19 11:23:24 +08:00 |
Frank Lee
|
ddcf58cacf
|
Revert "[sync] sync feature/shardformer with develop"
|
2023-06-09 09:41:27 +08:00 |
Frank Lee
|
eb39154d40
|
[dtensor] updated api and doc (#3845)
|
2023-06-08 10:18:17 +08:00 |
Frank Lee
|
80eba05b0a
|
[test] refactor tests with spawn (#3452)
* [test] added spawn decorator
* polish code
* polish code
* polish code
* polish code
* polish code
* polish code
|
2023-04-06 14:51:35 +08:00 |
YuliangLiu0306
|
b5a3a4a65f
|
[device] find best logical mesh
|
2023-01-05 17:21:29 +08:00 |
YuliangLiu0306
|
9c9246c0d9
|
[device] alpha beta profiler (#2311)
* [device] alpha beta profiler
* add usage
* fix variable name
|
2023-01-05 16:39:55 +08:00 |
Genghan Zhang
|
6630d45546
|
[autoparallel] Add alpha beta (#1973)
* Add alpha beta
* Fix test
* Fix test
|
2022-11-17 16:01:14 +08:00 |
YuliangLiu0306
|
b73fb7a077
|
[tensor] support runtime ShardingSpec apply (#1453)
* [tensor] support runtime ShardingSpec apply
* polish code
* polish code
|
2022-08-19 13:39:51 +08:00 |
YuliangLiu0306
|
0442f940f0
|
[device] add DeviceMesh class to support logical device layout (#1394)
* [device] add DeviceMesh class to support logical device layout
* polish code
* add doc string
|
2022-08-02 19:23:48 +08:00 |