YuliangLiu0306 aa0f6686f9 [autoparallel] accelerate gpt2 training (#2495)		2023-01-29 11:13:15 +08:00
..
deprecated	Revert "[NFC] polish code format" (#2372)	2023-01-06 16:01:09 +08:00
node_handler	[autoparallel] accelerate gpt2 training (#2495)	2023-01-29 11:13:15 +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] support origin activation ckpt on autoprallel system (#2468)	2023-01-16 16:25:13 +08:00
sharding_strategy.py	[autoparallel] memory estimation for shape consistency (#2144)	2022-12-21 10:39:37 +08:00