GitLab

Co-authored-by: github-actions <github-actions@github.com>

* [docker] fixed ninja build command

* polish code

* [checkpointio] unsharded optimizer checkpoint for Gemini plugin

* [checkpointio] unsharded optimizer checkpoint for Gemini using all_gather

Co-authored-by: github-actions <github-actions@github.com>

* [shardformer] made tensor parallelism configurable

* polish code

* [shardformer] refactored some doc and api

* polish code

* [shardformer] add benchmark of shardformer

* [shardformer] add benchmark of shardformer

* first v of vit shardformer

* keep vit

* update

* vit shard add vitattention vitlayer

* update num head shard para

* finish test for vit

* add new_model_class & postprocess

* add vit readme

* delete old files & fix the conflict

* fix sth

* [shardformer] shardformer support opt models

* [shardformer] shardformer support opt models, fix

* [shardformer] shardformer support opt models, fix

* [shardformer] shardformer support opt models, fix

* [test] fixed tests failed due to dtensor change

* polish code

* add layernorm to bert

* add layernorm test

* add layernorm test with load state dict

* add use_mixedfusedLN in shard config

* refactor policy to support fused_layernorm

* add linearconv1d test

* add linearconv1d test

* [shardformer] support module saving and loading

* polish code

* support kit use for bert test

* support kit test for gpt2

* [shardformer] adapted T5 and LLaMa test to use kit

* polish code

* add gpt2 test and layer class refactor

* add dropout in gpt2 policy

* fix bert downstream with new api

* remove comment line

* [shardformer] refactored embedding and dropout to parallel module

* polish code