main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\solver\graph_analysis.py
27 3 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit_utils.py
16 5 -
A
colossalai\legacy\nn\layer\vanilla\layers.py
38 5 -
A
tests\test_booster\test_mixed_precision\test_fp16_torch.py
7 3 -
A
tests\test_device\test_search_logical_device_mesh.py
4 3 -
A
tests\test_booster\test_plugin\test_dp_plugin_base.py
16 4 -
A
colossalai\legacy\nn\layer\wrapper\pipeline_wrapper.py
9 3 -
A
tests\test_booster\test_plugin\test_low_level_zero_plugin.py
22 3 -
A
colossalai\auto_parallel\tensor_shard\solver\strategies_constructor.py
29 12 -
A
tests\test_fp8\test_fp8_all_to_all.py
6 3 -
A
tests\test_fp8\test_all_to_all_single.py
7 2 -
A
tests\test_checkpoint_io\test_torch_ddp_checkpoint_io.py
5 3 -
A
tests\test_checkpoint_io\test_safetensors_async_io.py
13 5 -
A
.github\workflows\doc_test_on_pr.yml
 5 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
applications\Colossal-LLaMA\train_sft.example.sh
 1 -
A
applications\ColossalChat\coati\utils\ckpt_io.py
7 1 -
A
applications\ColossalQA\tests\test_memory.py
7 1 -
A
colossalai\fx\tracer\meta_patch\__init__.py
 3 -
A
colossalai\testing\random.py
3 2 -