main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\cuda_native\csrc\scaled_upper_triang_masked_softmax_cuda.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\type_shim.h
20 2 -
A
colossalai\kernel\cuda_native\layer_norm.py
9 1 -
A
colossalai\kernel\cuda_native\multihead_attention.py
31 1 -
A
colossalai\kernel\cuda_native\scaled_softmax.py
36 1 -
A
colossalai\kernel\jit\__init__.py
 1 -
A
colossalai\kernel\jit\bias_dropout_add.py
3 1 -
A
colossalai\kernel\jit\bias_gelu.py
4 1 -
A
colossalai\kernel\jit\option.py
5 1 -
A
colossalai\logging\__init__.py
5 1 -
A
colossalai\logging\logger.py
18 1 -
A
colossalai\nn\__init__.py
 1 -
A
colossalai\nn\_ops\__init__.py
 3 -
A
colossalai\nn\_ops\_utils.py
46 3 -
A
colossalai\nn\_ops\addmm.py
12 3 -
A
colossalai\nn\_ops\element_wise.py
1 1 -
A
colossalai\nn\_ops\embedding.py
9 3 -
A
colossalai\nn\_ops\embedding_bag.py
7 3 -
A
colossalai\nn\_ops\layernorm.py
1 3 -
A
colossalai\nn\_ops\linear.py
30 3 -