main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\logging\logger.py
18 1 -
A
colossalai\logging\__init__.py
5 1 -
A
colossalai\kernel\jit\option.py
5 1 -
A
colossalai\kernel\jit\bias_gelu.py
4 1 -
A
colossalai\kernel\jit\bias_dropout_add.py
3 1 -
A
colossalai\kernel\jit\__init__.py
 1 -
A
colossalai\kernel\cuda_native\scaled_softmax.py
36 1 -
A
colossalai\kernel\cuda_native\multihead_attention.py
31 1 -
A
colossalai\kernel\cuda_native\layer_norm.py
9 1 -
A
colossalai\kernel\cuda_native\csrc\type_shim.h
20 2 -
A
colossalai\kernel\cuda_native\csrc\scaled_upper_triang_masked_softmax_cuda.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\scaled_upper_triang_masked_softmax.h
94 2 5
A
colossalai\kernel\cuda_native\csrc\scaled_upper_triang_masked_softmax.cpp
3 1 -
A
colossalai\kernel\cuda_native\csrc\scaled_masked_softmax_cuda.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\scaled_masked_softmax.h
93 2 4
A
colossalai\kernel\cuda_native\csrc\scaled_masked_softmax.cpp
4 3 -
A
colossalai\kernel\cuda_native\csrc\multihead_attention_1d.h
9 1 -
A
colossalai\kernel\cuda_native\csrc\multihead_attention_1d.cpp
28 1 -
A
colossalai\kernel\cuda_native\csrc\multi_tensor_sgd_kernel.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\multi_tensor_scale_kernel.cu
 1 -