main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_shardformer\test_model\test_shard_falcon.py
20 1 -
A
tests\test_shardformer\test_model\test_shard_chatglm2.py
20 2 -
A
tests\test_shardformer\test_model\test_shard_bloom.py
20 2 -
A
tests\test_shardformer\test_model\test_shard_blip2.py
5 1 -
A
tests\test_shardformer\test_model\test_shard_bert.py
20 2 -
A
tests\test_shardformer\test_model\_utils.py
61 3 -
A
tests\test_shardformer\test_layer\test_vocab_parallel_embedding_1d.py
5 1 -
A
tests\test_shardformer\test_layer\test_qkv_fused_linear_1d.py
13 1 -
A
tests\test_shardformer\test_layer\test_linear_1d.py
18 1 -
A
tests\test_shardformer\test_layer\test_layernorm.py
5 1 -
A
tests\test_shardformer\test_layer\test_gpt2_qkv_fused_linear_1d.py
16 1 -
A
tests\test_shardformer\test_layer\test_embedding.py
5 1 -
A
tests\test_shardformer\test_layer\test_dropout.py
10 1 -
A
tests\test_shardformer\test_layer\test_dist_crossentropy.py
2 2 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_zero_optimizer.py
22 3 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_naive_optimizer.py
25 3 1
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_amp_optimizer.py
30 3 1
A
tests\test_pipeline\test_stage_manager.py
8 1 -
A
tests\test_pipeline\test_schedule\test_pipeline_schedule_utils.py
3 1 -
A
tests\test_pipeline\test_schedule\test_oneF_oneB.py
16 1 -