main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\cuda_native\csrc\kernels\include\cross_entropy_layer.h
2 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\context.h
6 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\block_reduce.h
5 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\general_kernels.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\dropout_kernels.cu
 2 -
A
colossalai\kernel\cuda_native\csrc\kernels\cuda_util.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\cublas_wrappers.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\cross_entropy.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\cpu_adam.h
10 2 -
A
colossalai\kernel\cuda_native\csrc\cpu_adam.cpp
80 2 3
A
colossalai\kernel\cuda_native\csrc\compat.h
0 1 -
A
colossalai\kernel\cuda_native\csrc\colossal_C_frontend.cpp
1 1 -
A
colossalai\kernel\cuda_native\__init__.py
 1 -
A
colossalai\kernel\__init__.py
 1 -
A
colossalai\initialize.py
64 2 1
A
colossalai\global_variables.py
5 1 -
A
colossalai\gemini\tensor_utils.py
18 1 -
A
colossalai\gemini\tensor_placement_policy.py
24 1 -
A
colossalai\gemini\stateful_tensor_mgr.py
18 1 -
A
colossalai\gemini\stateful_tensor.py
33 1 -