mirror of https://github.com/hpcaitech/ColossalAI
92f6791095
* 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 |
||
---|---|---|
.. | ||
__init__.py | ||
shard_config.py | ||
sharder.py | ||
shardformer.py |