Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-08 03:34:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1ee5875d61882fbf899e7935c04aa46cd60566c4
ColossalAI/colossalai/kernel/triton
History
pre-commit-ci[bot] 1ee5875d61 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2026-05-25 17:39:10 +00:00
..
__init__.py
[Infer] Revise and Adapt Triton Kernels for Spec-Dec (#5401)
2024-04-10 11:07:51 +08:00
context_attn_unpad.py
[kernel] Support New KCache Layout - Triton Kernel (#5677)
2024-05-03 17:20:45 +08:00
flash_decoding.py
[NFC] Fix code factors on inference triton kernels (#5743)
2024-05-21 22:12:15 +08:00
fused_rotary_embedding.py
[Inference]Fused the gate and up proj in mlp,and optimized the autograd process. (#5365)
2024-02-06 19:38:25 +08:00
kvcache_copy.py
[kernel] Support New KCache Layout - Triton Kernel (#5677)
2024-05-03 17:20:45 +08:00
llama_act_combine_kernel.py
[pre-commit.ci] auto fixes from pre-commit.com hooks
2026-05-25 17:39:10 +00:00
no_pad_rotary_embedding.py
[kernel] Support New KCache Layout - Triton Kernel (#5677)
2024-05-03 17:20:45 +08:00
qkv_matmul_kernel.py
[misc] update pre-commit and run all files (#4752)
2023-09-19 14:20:26 +08:00
rms_layernorm.py
[fix] multi graphs capture error
2024-03-11 10:49:31 +08:00
rotary_cache_copy.py
[Inference]Fused the gate and up proj in mlp,and optimized the autograd process. (#5365)
2024-02-06 19:38:25 +08:00
softmax.py
[misc] update pre-commit and run all files (#4752)
2023-09-19 14:20:26 +08:00
Powered by Gitea Version: 1.25.2 Page: 984ms Template: 75ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API