main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_shardformer\test_flash_attention.py
19 3 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_amp_optimizer.py
30 4 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_naive_optimizer.py
25 4 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_zero_optimizer.py
22 5 -
A
tests\test_shardformer\test_layer\test_dist_crossentropy.py
2 5 -
A
tests\test_shardformer\test_layer\test_dist_log_prob.py
0 1 -
A
tests\test_shardformer\test_layer\test_dropout.py
10 3 -
A
tests\test_shardformer\test_layer\test_embedding.py
5 3 -
A
tests\test_shardformer\test_layer\test_gpt2_qkv_fused_linear_1d.py
15 4 -
A
tests\test_shardformer\test_layer\test_layernorm.py
5 3 -
A
tests\test_shardformer\test_layer\test_linear_1d.py
24 2 -
A
tests\test_shardformer\test_layer\test_qkv_fused_linear_1d.py
10 3 -
A
tests\test_shardformer\test_layer\test_ring_attn.py
12 2 -
A
tests\test_shardformer\test_layer\test_sequence_parallel.py
13 2 -
A
tests\test_shardformer\test_layer\test_vocab_parallel_embedding_1d.py
5 4 -
A
tests\test_shardformer\test_model\_utils.py
73 5 -
A
tests\test_shardformer\test_model\test_shard_bert.py
20 7 -
A
tests\test_shardformer\test_model\test_shard_blip2.py
5 5 -
A
tests\test_shardformer\test_model\test_shard_bloom.py
20 6 -
A
tests\test_shardformer\test_model\test_shard_chatglm2.py
21 9 -