Commit Graph

1 Commits (1c7df566e23d5b94512f0777e0475df0a0ae1072)

Author SHA1 Message Date
LuGY a78daf6180
[shardformer] support interleaved pipeline (#4448)
* support interleaved pipeline

* fix unit test

* remove virtual stage test in stage mgr

* add droped type hint and updated bwd
2023-08-16 19:29:03 +08:00