main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_legacy\test_layers\test_3d\checks_3d\check_layer_3d.py
13 4 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_unet.py
2 7 -
A
colossalai\legacy\nn\layer\parallel_2p5d\__init__.py
 2 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
applications\ColossalChat\tests\test_lora.py
7 3 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit.py
2 3 -
A
colossalai\legacy\nn\layer\parallel_2p5d\_utils.py
3 3 -
A
tests\test_booster\test_accelerator.py
1 5 -
A
tests\test_booster\test_plugin\test_torch_ddp_plugin.py
19 8 -
A
tests\test_shardformer\test_model\test_shard_llama.py
38 4 -
A
tests\test_shardformer\test_model\test_shard_gptj.py
20 4 -
A
tests\test_legacy\test_layers\test_sequence\test_sequence.py
4 3 -
A
tests\test_legacy\test_layers\test_1d\checks_1d\common.py
1 1 -
A
tests\test_legacy\test_comm\test_comm.py
8 5 -
A
tests\test_infer\test_models\test_custom_model.py
13 1 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_opt.py
2 13 -
A
tests\test_device\test_search_logical_device_mesh.py
4 3 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -