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
|
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 |
Jiarui Fang
|
9f4fb3f28a
|
[ColoTensor] ColoInitContext initialize parameters in shard mode. (#1937)
|
2022-11-14 16:05:09 +08:00 |
Frank Lee
|
eee84908d4
|
[autoparallel] handled illegal sharding strategy (#1728)
* [autoparallel] handled illegal sharding strategy
* polish code
|
2022-10-19 12:53:06 +08:00 |
Frank Lee
|
ae1b58cd16
|
[tensor] added linear implementation for the new sharding spec (#1416)
* [tensor] added linear implementation for the new sharding spec
* polish code
|
2022-08-12 11:33:09 +08:00 |