Genghan Zhang d655eea515 [autoparallel] mix gather (#1977) ...

* Add mix-gather

* Add comments

* Add comments

* Polish comments

* Change the global rank assumption

* Add tests

* Add two-step tests

* Fix 10 and 01

* Skip test becasue the number of GPUs

2022-11-23 21:49:17 +08:00

12 KiB

Raw Blame History

View Raw