mirror of https://github.com/hpcaitech/ColossalAI
1fb0d95df0
* [shardformer] made tensor parallelism configurable * polish code |
||
---|---|---|
.. | ||
test_layer | ||
test_model | ||
__init__.py | ||
test_with_torch_ddp.py |