You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

flybird11111 5e16bf7980 [shardformer] fix gathering output when using tensor parallelism (#5431) ... * fix * padding vocab_size when using pipeline parallellism padding vocab_size when using pipeline parallellism fix fix * fix * fix fix fix * fix gather output * fix * fix * fix fix resize embedding fix resize embedding * fix resize embedding fix * revert * revert * revert		9 months ago
..
__init__.py	[misc] update pre-commit and run all files (#4752)	1 year ago
shard_config.py	[shardformer] fix gathering output when using tensor parallelism (#5431)	9 months ago
sharder.py	[nfc] fix typo colossalai/shardformer/ (#5133)	11 months ago
shardformer.py	[example]add gpt2 benchmark example script. (#5295)	9 months ago
utils.py	[pipeline] update shardformer policy	1 year ago