main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\pybind\cpu_adam\cpu_adam_arm.py
7 1 -
A
extensions\pybind\cpu_adam\cpu_adam_x86.py
7 1 -
A
extensions\pybind\flash_attention\__init__.py
 1 -
A
extensions\pybind\flash_attention\flash_attention_dao_cuda.py
7 2 -
A
extensions\pybind\flash_attention\flash_attention_npu.py
7 2 -
A
extensions\pybind\flash_attention\flash_attention_sdpa_cuda.py
7 1 -
A
extensions\pybind\inference\__init__.py
 1 -
A
extensions\pybind\inference\inference.cpp
1 1 -
A
extensions\pybind\inference\inference_ops_cuda.py
5 1 -
A
extensions\pybind\layernorm\__init__.py
 1 -
A
extensions\pybind\layernorm\layer_norm.cpp
13 1 -
A
extensions\pybind\layernorm\layernorm_cuda.py
6 1 -
A
extensions\pybind\moe\__init__.py
 1 -
A
extensions\pybind\moe\moe.cpp
6 1 -
A
extensions\pybind\moe\moe_cuda.py
5 1 -
A
extensions\pybind\optimizer\__init__.py
 1 -
A
extensions\pybind\optimizer\fused_optimizer_cuda.py
5 1 -
A
extensions\pybind\optimizer\optimizer.cpp
1 1 -
A
extensions\pybind\softmax\__init__.py
 1 -
A
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -