mirror of https://github.com/hpcaitech/ColossalAI
![]() * 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 |
||
---|---|---|
.. | ||
test_layer | ||
test_model | ||
__init__.py |