main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_view_handler.py
22 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_where_handler.py
5 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\utils.py
26 14 -
A
tests\test_auto_parallel\test_tensor_shard\test_solver_with_resnet_v2.py
5 9 -
A
tests\test_autochunk\test_autochunk_alphafold\benchmark_autochunk_alphafold.py
29 5 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_alphafold_utils.py
22 4 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_evoformer_block.py
4 4 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_evoformer_stack.py
3 3 -
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 -