Frank Lee
|
c4b1b65931
|
[test] fixed tests failed due to dtensor change (#4082)
* [test] fixed tests failed due to dtensor change
* polish code
|
1 year ago |
Frank Lee
|
8eb09a4c69
|
[shardformer] support module saving and loading (#4062)
* [shardformer] support module saving and loading
* polish code
|
1 year ago |
digger yu
|
e61ffc77c6
|
fix typo tests/ (#3936)
|
1 year ago |
Frank Lee
|
ddcf58cacf
|
Revert "[sync] sync feature/shardformer with develop"
|
1 year ago |
Frank Lee
|
eb39154d40
|
[dtensor] updated api and doc (#3845)
|
1 year ago |
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
|
2 years ago |
YuliangLiu0306
|
2eca4cd376
|
[DTensor] refactor dtensor with new components (#3089)
* [DTensor] refactor dtensor with new components
* polish
|
2 years ago |
YuliangLiu0306
|
8e4e8601b7
|
[DTensor] implement layout converter (#3055)
* [DTensor] refactor LayoutConverter for DTensor
* polish code
* polish docstring
|
2 years ago |
YuliangLiu0306
|
29386a54e6
|
[DTensor] refactor CommSpec (#3034)
|
2 years ago |
YuliangLiu0306
|
4269196c79
|
[hotfix] skip auto checkpointing tests (#3029)
* [hotfix] skip auto checkpointing tests
* fix test name issue
|
2 years ago |
YuliangLiu0306
|
cd2b0eaa8d
|
[DTensor] refactor sharding spec (#2987)
* [autoparallel] refactor sharding spec
* rename function name
|
2 years ago |