main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_infer\test_kernels\triton\test_context_attn_unpad.py
12 2 -
A
examples\tutorial\large_batch_optimizer\train.py
12 4 -
A
colossalai\auto_parallel\passes\comm_metainfo_pass.py
12 3 -
A
tests\test_moe\moe_utils.py
12 10 -
A
colossalai\legacy\zero\gemini\ophooks\_shard_param_ophook.py
12 2 -
A
applications\ColossalChat\coati\distributed\loss.py
12 1 -
A
colossalai\legacy\inference\pipeline\benchmark\benchmark.py
12 2 -
A
colossalai\legacy\inference\tensor_parallel\kvcache_manager.py
12 2 -
A
tests\test_legacy\test_utils\test_checkpoint\test_checkpoint_2p5d.py
12 3 -
A
tests\test_legacy\test_utils\test_checkpoint\test_checkpoint_2d.py
12 3 -
A
tests\test_legacy\test_utils\test_checkpoint\test_checkpoint_1d.py
12 3 -
A
tests\test_legacy\test_utils\test_checkpoint\test_checkpoint_3d.py
12 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\permute_handler.py
12 2 -
A
tests\test_infer\test_kernels\triton\test_kvcache_copy.py
12 1 -
A
colossalai\legacy\tensor\distspec.py
12 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\sum_handler.py
12 2 -
A
tests\test_shardformer\test_layer\test_ring_attn.py
12 2 -
A
colossalai\legacy\nn\parallel\layers\colo_module.py
12 3 -
A
colossalai\legacy\registry\registry.py
12 2 -
A
colossalai\booster\plugin\plugin_base.py
12 3 -