ColossalAI/colossalai/shardformer/policies
flybird11111 de8a65babc
[shardformer] opt fix. (#4514)
* [shardformer] chatglm support sequence parallel

[shardformer] chatglm support sequence parallel

[shardformer] chatglm support sequence parallel

[shardformer] chatglm support sequence parallel

[shardformer] chatglm support sequence parallel

[shardformer] chatglm support sequence parallel

* fix

fix

fix

fix

* [shardformer] jit fused fix

* [shardformer] jit fused fix

* [shardformer] jit fused fix

* [shardformer] jit fused fix

* [shardformer] jit fused fix

* [shardformer] jit fused fix

* [shardformer] jit fused fix

* activate checks

* [Test] test ci

* test ci

* test ci

* test ci

* test ci

* test ci

* test ci

* fix
2023-08-25 19:41:24 +08:00
..
__init__.py
auto_policy.py rename chatglm to chatglm2 (#4484) 2023-08-22 14:13:31 +08:00
base_policy.py
bert.py [shardformer] chatglm support sequence parallel (#4482) 2023-08-22 23:59:31 +08:00
blip2.py [shardformer] chatglm support sequence parallel (#4482) 2023-08-22 23:59:31 +08:00
bloom.py [shardformer] chatglm support sequence parallel (#4482) 2023-08-22 23:59:31 +08:00
chatglm2.py [shardformer] chatglm support sequence parallel (#4482) 2023-08-22 23:59:31 +08:00
gpt2.py [shardformer] chatglm support sequence parallel (#4482) 2023-08-22 23:59:31 +08:00
llama.py [shardformer] vit/llama/t5 ignore the sequence parallelism flag and some fix. (#4498) 2023-08-24 15:50:02 +08:00
opt.py [shardformer] opt fix. (#4514) 2023-08-25 19:41:24 +08:00
sam.py [shardformer] chatglm support sequence parallel (#4482) 2023-08-22 23:59:31 +08:00
t5.py [shardformer] vit/llama/t5 ignore the sequence parallelism flag and some fix. (#4498) 2023-08-24 15:50:02 +08:00
vit.py [shardformer] vit/llama/t5 ignore the sequence parallelism flag and some fix. (#4498) 2023-08-24 15:50:02 +08:00
whisper.py [shardformer] vit/llama/t5 ignore the sequence parallelism flag and some fix. (#4498) 2023-08-24 15:50:02 +08:00