ColossalAI/colossalai/kernel/triton
2024-06-14 16:27:46 +08:00
..
__init__.py [Inference] Dynamic Batching Inference, online and offline (#4953) 2023-10-30 10:52:19 +08:00
context_attention.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-14 16:27:46 +08:00
copy_kv_cache_dest.py [Inference] Dynamic Batching Inference, online and offline (#4953) 2023-10-30 10:52:19 +08:00
custom_autotune.py add autotune (#4822) 2023-09-28 13:47:35 +08:00
flash_decoding.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-14 16:27:46 +08:00
fused_layernorm.py
gptq_triton.py [inference] add reference and fix some bugs (#4937) 2023-10-20 13:39:34 +08:00
int8_rotary_embedding_kernel.py [inference] Add smmoothquant for llama (#4904) 2023-10-16 11:28:44 +08:00
llama_act_combine_kernel.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-14 16:27:46 +08:00
qkv_matmul_kernel.py
self_attention_nofusion.py [Refactor] Integrated some lightllm kernels into token-attention (#4946) 2023-10-19 22:22:47 +08:00
smooth_attention.py [inference] add reference and fix some bugs (#4937) 2023-10-20 13:39:34 +08:00
softmax.py
token_attention_kernel.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-14 16:27:46 +08:00