main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\pybind\moe\moe_cuda.py
5 1 -
A
extensions\pybind\moe\moe.cpp
6 1 -
A
extensions\pybind\moe\__init__.py
 1 -
A
extensions\pybind\layernorm\layernorm_cuda.py
6 1 -
A
extensions\pybind\layernorm\layer_norm.cpp
13 1 -
A
extensions\pybind\layernorm\__init__.py
 1 -
A
extensions\pybind\inference\inference_ops_cuda.py
5 1 -
A
extensions\pybind\inference\inference.cpp
1 1 -
A
extensions\pybind\inference\__init__.py
 1 -
A
extensions\pybind\flash_attention\flash_attention_sdpa_cuda.py
7 1 -
A
extensions\pybind\flash_attention\flash_attention_npu.py
7 1 -
A
extensions\pybind\flash_attention\flash_attention_dao_cuda.py
7 2 -
A
extensions\pybind\flash_attention\__init__.py
 1 -
A
extensions\pybind\cpu_adam\cpu_adam_x86.py
7 1 -
A
extensions\pybind\cpu_adam\cpu_adam_arm.py
7 1 -
A
extensions\pybind\cpu_adam\__init__.py
 1 -
A
extensions\cuda_extension.py
14 8 -
A
extensions\csrc\kernel\x86\cpu_adam.h
16 2 -
A
extensions\csrc\kernel\x86\cpu_adam.cpp
66 1 -
A
extensions\csrc\kernel\cuda\utils\vec_copy.h
7 1 -