main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_analyzer\test_subclasses\test_flop_tensor.py
3 5 -
A
tests\test_auto_parallel\test_offload\test_perf.py
8 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_compatibility_with_gemini.py
11 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\registry.py
4 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_liveness_analysis.py
3 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_matmul_metainfo.py
1 5 -
A
tests\test_fx\test_pipeline\test_hf_model\test_gpt.py
2 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_matmul_handler.py
18 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_shard_option.py
9 5 -
A
tests\test_cluster\test_process_group_mesh.py
7 5 -
A
colossalai\legacy\nn\layer\parallel_1d\_operation.py
9 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_tensor_constructor.py
5 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_where_handler.py
5 5 -
A
colossalai\legacy\nn\layer\parallel_1d\layers.py
 5 1
A
tests\test_autochunk\test_autochunk_alphafold\benchmark_autochunk_alphafold.py
29 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\normal_pooling_generator.py
10 5 -
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 5 4
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\output_generator.py
15 5 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_transformer_utils.py
31 5 -