main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\jit\__init__.py
 2 -
A
colossalai\kernel\jit\bias_dropout_add.py
3 3 -
A
colossalai\kernel\jit\bias_gelu.py
4 6 -
A
colossalai\kernel\jit\option.py
10 5 -
A
colossalai\kernel\kernel_loader.py
7 5 -
A
colossalai\kernel\triton\__init__.py
 11 -
A
colossalai\kernel\triton\context_attn_unpad.py
26 1 -
A
colossalai\kernel\triton\flash_decoding.py
18 3 -
A
colossalai\kernel\triton\fused_rotary_embedding.py
3 1 -
A
colossalai\kernel\triton\kvcache_copy.py
14 1 -
A
colossalai\kernel\triton\llama_act_combine_kernel.py
9 2 -
A
colossalai\kernel\triton\no_pad_rotary_embedding.py
15 1 -
A
colossalai\kernel\triton\qkv_matmul_kernel.py
3 2 -
A
colossalai\kernel\triton\rms_layernorm.py
10 1 -
A
colossalai\kernel\triton\rotary_cache_copy.py
6 1 -
A
colossalai\kernel\triton\softmax.py
10 2 -
A
colossalai\lazy\__init__.py
 2 -
A
colossalai\lazy\construction.py
10 1 -
A
colossalai\lazy\lazy_init.py
87 3 -
A
colossalai\lazy\pretrained.py
48 3 -