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-04-11 06:33:30 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
fded91d049997ed87dee965fc42c35a239e3ec03
ColossalAI/colossalai/kernel/triton
History
Jianghai fded91d049 [Inference] Kernel: no pad rotary embedding (#5252)
* fix bugs

* comment

* use more accurate atol

* fix
2024-01-11 13:46:14 +00:00
..
__init__.py
[Inference] Kernel: no pad rotary embedding (#5252)
2024-01-11 13:46:14 +00:00
context_attn_unpad.py
[kernel] Add triton kernel for context attention (FAv2) without padding (#5192)
2024-01-11 13:39:56 +00:00
custom_autotune.py
add autotune (#4822)
2023-09-28 13:47:35 +08:00
fused_layernorm.py
[misc] update pre-commit and run all files (#4752)
2023-09-19 14:20:26 +08:00
gptq_triton.py
[inference] add reference and fix some bugs (#4937)
2023-10-20 13:39:34 +08:00
llama_act_combine_kernel.py
[moe] merge moe into main (#4978)
2023-11-02 02:21:24 +00:00
no_pad_rotary_embedding.py
[Inference] Kernel: no pad rotary embedding (#5252)
2024-01-11 13:46:14 +00:00
qkv_matmul_kernel.py
[misc] update pre-commit and run all files (#4752)
2023-09-19 14:20:26 +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: 281ms Template: 4ms
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