main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_fx\test_comm_size_compute.py
4 9 -
A
colossalai\fx\passes\shard_1d_pass.py
12 9 -
A
colossalai\shardformer\layer\_operation.py
189 9 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\__init__.py
 9 -
A
colossalai\pipeline\schedule\generate.py
74 9 -
A
tests\test_auto_parallel\test_tensor_shard\test_solver_with_resnet_v2.py
5 9 -
A
colossalai\auto_parallel\tensor_shard\node_handler\matmul_handler.py
62 9 -
A
colossalai\shardformer\layer\attn.py
107 9 -
A
colossalai\checkpoint_io\checkpoint_io_base.py
32 8 -
A
colossalai\utils\model\utils.py
18 8 -
A
colossalai\shardformer\layer\linear.py
85 8 -
A
colossalai\checkpoint_io\moe_checkpoint.py
155 8 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_output_handler.py
6 8 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_norm_pooling_handler.py
3 8 -
A
tests\kit\model_zoo\transformers\llama.py
2 8 -
A
colossalai\inference\__init__.py
 8 -
A
tests\kit\model_zoo\registry.py
12 8 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\norm.py
2 8 -
A
colossalai\auto_parallel\tensor_shard\utils\factory.py
25 8 -
A
colossalai\autochunk\estimate_memory.py
75 8 -