Baizhou Zhang 208ac8f2ba [pipeline] Add Pipeline Forward for GPT2Model Shardformer (#4224) ... * * fix typehint & docstring in sharder.py * * update pipeline forward for GPT2Model * * add test for pipeline forward of GPT2Model * * add cache cleaning in gpt2 test * * change assert to raise command		2023-08-15 23:25:14 +08:00
..
__init__.py	[shardformer] Refactor shardformer api (#4001)	2023-07-04 16:05:01 +08:00
shard_config.py	[shardformer] fix type hint	2023-08-15 23:25:14 +08:00
sharder.py	[pipeline] Add Pipeline Forward for GPT2Model Shardformer (#4224)	2023-08-15 23:25:14 +08:00
shardformer.py	[shardformer] rename policy file name	2023-08-15 23:25:14 +08:00
utils.py	[pipeline] update shardformer policy	2023-08-15 23:25:14 +08:00