mirror of https://github.com/hpcaitech/ColossalAI
1fb0d95df0
* [shardformer] made tensor parallelism configurable * polish code |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _utils.py | ||
| test_shard_bert.py | ||
| test_shard_bloom.py | ||
| test_shard_gpt2.py | ||
| test_shard_llama.py | ||
| test_shard_opt.py | ||
| test_shard_t5.py | ||
| test_shard_vit.py | ||