main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_extramsa_block.py
3 4 -
A
tests\test_autochunk\test_autochunk_diffuser\benchmark_autochunk_diffuser.py
31 3 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_diffuser_utils.py
24 6 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_unet.py
2 7 -
A
tests\test_autochunk\test_autochunk_transformer\benchmark_autochunk_transformer.py
31 4 -
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 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit.py
2 3 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit_utils.py
16 5 -
A
tests\test_booster\test_accelerator.py
1 5 -
A
tests\test_booster\test_mixed_precision\test_fp16_torch.py
7 3 -
A
tests\test_booster\test_plugin\test_3d_plugin.py
29 3 -
A
tests\test_booster\test_plugin\test_dp_plugin_base.py
16 4 -
A
tests\test_booster\test_plugin\test_gemini_plugin.py
24 2 -
A
tests\test_booster\test_plugin\test_low_level_zero_plugin.py
22 3 -
A
tests\test_booster\test_plugin\test_torch_ddp_plugin.py
19 8 -
A
tests\test_booster\test_plugin\test_torch_fsdp_plugin.py
11 6 -
A
tests\test_checkpoint_io\test_gemini_checkpoint_io.py
25 5 -
A
tests\test_checkpoint_io\test_gemini_torch_compability.py
20 7 -
A
tests\test_checkpoint_io\test_general_checkpoint_io.py
8 2 -