/
githubmirror
/
ColossalAI
ОбзорДокументацияВойти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
ДокументацияПоддержка
Политика конфиденциальностиПользовательское соглашениеПолитика использования «cookies»Согласие субъекта персональных данных
2026 ©
ColossalAI/
.../kernel/triton/
..
__init__.py

[Infer] Revise and Adapt Triton Kernels for Spec-Dec (#5401)

2 года назад
context_attn_unpad.py

[kernel] Support New KCache Layout - Triton Kernel (#5677)

2 года назад
flash_decoding.py

[NFC] Fix code factors on inference triton kernels (#5743)

2 года назад
fused_rotary_embedding.py

[Inference]Fused the gate and up proj in mlp,and optimized the autograd process. (#5365)

3 года назад
kvcache_copy.py

[kernel] Support New KCache Layout - Triton Kernel (#5677)

2 года назад
llama_act_combine_kernel.py

[devops] remove post commit ci (#5566)

2 года назад
no_pad_rotary_embedding.py

[kernel] Support New KCache Layout - Triton Kernel (#5677)

2 года назад
qkv_matmul_kernel.py

[misc] update pre-commit and run all files (#4752)

3 года назад
rms_layernorm.py

[fix] multi graphs capture error

2 года назад
rotary_cache_copy.py

[Inference]Fused the gate and up proj in mlp,and optimized the autograd process. (#5365)

3 года назад
softmax.py

[misc] update pre-commit and run all files (#4752)

3 года назад