main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\layer_norm_handler.py
3 3 -
A
extensions\csrc\kernel\cuda\scaled_upper_triang_masked_softmax_kernel.cu
 1 -
A
colossalai\legacy\inference\quant\smoothquant\models\base_model.py
73 1 -
A
extensions\csrc\kernel\cuda\utils\micros.h
0 1 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\__init__.py
 1 -
A
extensions\csrc\kernel\cuda\utils\vec_copy.h
7 1 -
A
colossalai\legacy\inference\quant\smoothquant\models\llama.py
92 1 -
A
extensions\csrc\kernel\x86\cpu_adam.h
16 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\matmul_handler.py
62 9 -
A
extensions\pybind\softmax\scaled_masked_softmax_cuda.py
5 1 -
A
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -
A
extensions\pybind\inference\inference.cpp
1 1 -
A
extensions\pybind\inference\__init__.py
 1 -
A
.github\workflows\doc_test_on_pr.yml
 5 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
applications\Colossal-LLaMA\train_sft.example.sh
 1 -
A
applications\ColossalChat\coati\utils\ckpt_io.py
7 1 -
A
applications\ColossalQA\tests\test_memory.py
7 1 -
A
colossalai\fx\tracer\meta_patch\__init__.py
 3 -
A
colossalai\utils\__init__.py
 2 -