Boyuan Yao d45695d94e Merge pull request #2258 from hpcaitech/debug/ckpt-autoparallel ... [autockpt] provide option for activation checkpoint search in SPMD solver		2023-01-04 11:37:28 +08:00
..
deprecated	[autoparallel] fix spelling error (#2270)	2023-01-03 16:13:00 +08:00
node_handler	Merge pull request #2258 from hpcaitech/debug/ckpt-autoparallel	2023-01-04 11:37:28 +08:00
solver	[autoparallel] fix spelling error (#2270)	2023-01-03 16:13:00 +08:00
utils	[autoparallel] adapt solver with self attention (#2037)	2022-12-01 17:53:15 +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] gpt2 autoparallel examples (#2267)	2023-01-03 14:23:33 +08:00
sharding_strategy.py	[autoparallel] memory estimation for shape consistency (#2144)	2022-12-21 10:39:37 +08:00