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
|
8eb09a4c69
|
[shardformer] support module saving and loading (#4062)
* [shardformer] support module saving and loading
* polish code
|
2023-07-04 16:05:01 +08:00 |
digger yu
|
e61ffc77c6
|
fix typo tests/ (#3936)
|
2023-06-09 09:49:41 +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
|
2eca4cd376
|
[DTensor] refactor dtensor with new components (#3089)
* [DTensor] refactor dtensor with new components
* polish
|
2023-03-14 16:25:47 +08:00 |
YuliangLiu0306
|
8e4e8601b7
|
[DTensor] implement layout converter (#3055)
* [DTensor] refactor LayoutConverter for DTensor
* polish code
* polish docstring
|
2023-03-10 09:53:52 +08:00 |
YuliangLiu0306
|
29386a54e6
|
[DTensor] refactor CommSpec (#3034)
|
2023-03-08 10:45:31 +08:00 |
YuliangLiu0306
|
4269196c79
|
[hotfix] skip auto checkpointing tests (#3029)
* [hotfix] skip auto checkpointing tests
* fix test name issue
|
2023-03-07 15:50:00 +08:00 |
YuliangLiu0306
|
cd2b0eaa8d
|
[DTensor] refactor sharding spec (#2987)
* [autoparallel] refactor sharding spec
* rename function name
|
2023-03-07 11:08:11 +08:00 |