mirror of https://github.com/hpcaitech/ColossalAI
108e54a0b4
* [shardformer] gpt2 tests fix [shardformer] test all optimizations (#4399) [shardformer] test all optimizations [shardformer] test all optimizations [shardformer] test all optimizations [shardformer] gpt2 tests fix * [shardformer]update t5 to use all optimizations |
||
---|---|---|
.. | ||
__init__.py | ||
_utils.py | ||
test_pure_pipeline.py | ||
test_shard_bert.py | ||
test_shard_blip2.py | ||
test_shard_bloom.py | ||
test_shard_chatglm.py | ||
test_shard_gpt2.py | ||
test_shard_llama.py | ||
test_shard_opt.py | ||
test_shard_sam.py | ||
test_shard_t5.py | ||
test_shard_vit.py | ||
test_shard_whisper.py |