main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_fx\test_pipeline\test_hf_model\test_bert.py
2 5 -
A
colossalai\legacy\engine\gradient_handler\_data_parallel_gradient_handler.py
2 4 -
A
examples\language\commons\utils.py
2 1 -
A
colossalai\legacy\nn\metric\__init__.py
3 2 -
A
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_function\linear.py
3 2 -
A
applications\ColossalQA\tests\test_document_loader.py
3 2 -
A
extensions\pybind\softmax\scaled_upper_triang_masked_softmax.cpp
3 1 -
A
colossalai\legacy\nn\layer\parallel_2d\_utils.py
3 3 -
A
colossalai\_analyzer\fx\symbolic_profile.py
3 2 -
A
colossalai\legacy\nn\metric\accuracy_2d.py
3 2 -
A
colossalai\inference\server\utils.py
3 1 -
A
tests\test_device\test_init_logical_pg.py
3 4 -
A
tests\test_device\test_alpha_beta.py
3 3 -
A
tests\test_checkpoint_io\utils.py
3 2 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_extramsa_block.py
3 4 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_evoformer_stack.py
3 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_norm_pooling_handler.py
3 8 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_norm_metainfo.py
3 3 -
A
tests\test_analyzer\test_subclasses\test_flop_tensor.py
3 5 -