YuliangLiu0306 2731531bc2 [autoparallel] integrate device mesh initialization into autoparallelize (#2393) ... * [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		2023-01-11 14:03:49 +08:00
..
deprecated	Revert "[NFC] polish code format" (#2372)	2023-01-06 16:01:09 +08:00
node_handler	[autoparallel] add shard option (#2423)	2023-01-11 13:40:33 +08:00
solver	[autoparallel] fix spelling error (#2270)	2023-01-03 16:13:00 +08:00
utils	Revert "[NFC] polish code format" (#2372)	2023-01-06 16:01:09 +08:00
__init__.py	[autoparallel] init new folder structure (#1696)	2022-10-13 14:18:55 +08:00
constants.py	[autoparallel] adapt solver with self attention (#2037)	2022-12-01 17:53:15 +08:00
initialize.py	[autoparallel] integrate device mesh initialization into autoparallelize (#2393)	2023-01-11 14:03:49 +08:00
sharding_strategy.py	[autoparallel] memory estimation for shape consistency (#2144)	2022-12-21 10:39:37 +08:00