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 |
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 |
Hongxin Liu
|
9c88b6cbd1
|
[lazy] fix compatibility problem on torch 1.13 (#3911)
|
2023-06-07 11:10:12 +08:00 |
Hongxin Liu
|
dbb32692d2
|
[lazy] refactor lazy init (#3891)
* [lazy] remove old lazy init
* [lazy] refactor lazy init folder structure
* [lazy] fix lazy tensor deepcopy
* [test] update lazy init test
|
2023-06-05 14:20:47 +08:00 |