main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\passes\meta_info_prop.py
32 5 -
A
extensions\pybind\flash_attention\flash_attention_npu.py
7 2 -
A
colossalai\legacy\inference\quant\gptq\__init__.py
 1 -
A
extensions\pybind\inference\__init__.py
 1 -
A
applications\ColossalChat\examples\training_scripts\train_kto.sh
 1 -
A
extensions\pybind\inference\inference_ops_cuda.py
5 1 -
A
colossalai\legacy\inference\quant\gptq\cai_gptq\cai_quant_linear.py
57 1 -
A
extensions\pybind\layernorm\layer_norm.cpp
13 1 -
A
colossalai\auto_parallel\passes\runtime_preparation_pass.py
52 29 -
A
tests\test_infer\test_drafter.py
4 1 -
A
tests\test_infer\test_cuda_graph.py
8 1 -
A
tests\test_checkpoint_io\test_gemini_checkpoint_io.py
25 5 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_diffuser_utils.py
24 6 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_sum_handler.py
15 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_getitem_handler.py
9 3 -
A
tests\test_auto_parallel\test_offload\model_utils.py
8 3 -
A
tests\kit\model_zoo\diffusers\__init__.py
 1 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -