ziyuhuang123 d344313533 [NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/embedding_handler.py code style (#2725)		2023-02-15 16:31:40 +08:00
..
deprecated	[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/embedding_handler.py code style (#2725)	2023-02-15 16:31:40 +08:00
node_handler	[autoparallel] Patch meta information of `torch.nn.LayerNorm` (#2647)	2023-02-10 14:29:24 +08:00
solver	[autoparallel] adapt autoparallel tests with latest api (#2626)	2023-02-08 15:02:12 +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	add overlap option (#2613)	2023-02-08 15:02:31 +08:00
sharding_strategy.py	[autoparallel] memory estimation for shape consistency (#2144)	2022-12-21 10:39:37 +08:00