main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_device\test_init_logical_pg.py
3 4 -
A
colossalai\legacy\communication\ring.py
3 4 -
A
tests\test_fp8\test_fp8_all_to_all_single.py
3 2 -
A
colossalai\testing\pytest_wrapper.py
3 5 -
A
colossalai\inference\server\utils.py
3 1 -
A
colossalai\shardformer\modeling\jit.py
3 2 -
A
extensions\pybind\softmax\scaled_upper_triang_masked_softmax.cpp
3 1 -
A
colossalai\kernel\jit\bias_dropout_add.py
3 3 -
A
examples\inference\benchmark_ops\benchmark_xine_copy.py
3 1 -
A
colossalai\kernel\triton\fused_rotary_embedding.py
3 1 -
A
colossalai\kernel\triton\qkv_matmul_kernel.py
3 2 -
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
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_norm_metainfo.py
3 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_norm_pooling_handler.py
3 8 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -
A
colossalai\legacy\amp\torch_amp\__init__.py
3 2 -