You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_shardformer/test_layer
flybird11111 79718fae04
[shardformer] llama support DistCrossEntropy (#5176)
12 months ago
..
test_dist_crossentropy.py [shardformer] llama support DistCrossEntropy (#5176) 12 months ago
test_dropout.py [misc] update pre-commit and run all files (#4752) 1 year ago
test_embedding.py [misc] update pre-commit and run all files (#4752) 1 year ago
test_gpt2_qkv_fused_linear_1d.py [misc] update pre-commit and run all files (#4752) 1 year ago
test_layernorm.py [misc] update pre-commit and run all files (#4752) 1 year ago
test_linear_1d.py [misc] update pre-commit and run all files (#4752) 1 year ago
test_qkv_fused_linear_1d.py [misc] update pre-commit and run all files (#4752) 1 year ago
test_vocab_parallel_embedding_1d.py [misc] update pre-commit and run all files (#4752) 1 year ago