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_dist_crossentropy.py | ||
test_dropout.py | ||
test_embedding.py | ||
test_layernorm.py | ||
test_linear_1d.py | ||
test_linearconv_1d.py | ||
test_vocab_parallel_embedding_1d.py |