[autoparallel] integrate device mesh initialization into autoparallelize (#2393)

Browse Source

* [autoparallel] integrate device mesh initialization into autoparallelize

* add megatron solution

* update gpt autoparallel examples with latest api

* adapt beta value to fit the current computation cost

...

This commit is contained in:

YuliangLiu0306

2023-01-11 14:03:49 +08:00

committed by GitHub

parent c72c827e95

commit 2731531bc2

7 changed files with 64 additions and 51 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

BIN

examples/language/gpt/experiments/auto_parallel/saved_solution/solution_12_layers.pt Normal file

Binary file not shown.

BIN

examples/language/gpt/experiments/auto_parallel/saved_solution/solution_1_layers.pt Normal file

Binary file not shown.

BIN

examples/language/gpt/experiments/auto_parallel/saved_solution/solution_4_layers.pt Normal file

Binary file not shown.