main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\policies\auto_policy.py
6 3 -
A
tests\test_zero\test_low_level\test_mem_leak.py
6 1 -
A
tests\test_tensor\test_shape_consistency_apply.py
6 3 -
A
tests\kit\model_zoo\transformers\deepseek_v3.py
6 2 -
A
tests\test_legacy\test_comm\test_boardcast_send_recv_v2.py
6 4 -
A
tests\test_infer\test_continuous_batching.py
6 2 -
A
tests\kit\model_zoo\torchaudio\torchaudio.py
5 3 -
A
tests\kit\model_zoo\custom\simple_net.py
5 1 -
A
tests\kit\model_zoo\custom\hanging_param_model.py
5 1 -
A
extensions\pybind\softmax\scaled_upper_triangle_masked_softmax_cuda.py
5 1 -
A
tests\kit\model_zoo\transformers\albert.py
5 3 -
A
extensions\pybind\softmax\scaled_masked_softmax_cuda.py
5 1 -
A
extensions\pybind\moe\moe_cuda.py
5 1 -
A
extensions\pybind\optimizer\fused_optimizer_cuda.py
5 1 -
A
extensions\pybind\inference\inference_ops_cuda.py
5 1 -
A
colossalai\inference\graph_runner.py
5 1 -
A
colossalai\inference\modeling\layers\baichuan_tp_linear.py
5 3 -
A
tests\kit\model_zoo\transformers\bloom.py
5 6 -
A
tests\test_fx\test_pipeline\test_torchvision\test_torchvision.py
5 6 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\base.py
5 2 -