github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'cabc1286ca'
${ noResults }
ColossalAI/colossalai/kernel/triton
History
Xu Kai 785802e809
[inference] add reference and fix some bugs (#4937) 		1 year ago
..
__init__.py [Refactor] Integrated some lightllm kernels into token-attention (#4946) 1 year ago
context_attention.py [Refactor] Integrated some lightllm kernels into token-attention (#4946) 1 year ago
copy_kv_cache_dest.py [Refactor] Integrated some lightllm kernels into token-attention (#4946) 1 year ago
custom_autotune.py add autotune (#4822) 1 year ago
fused_layernorm.py [misc] update pre-commit and run all files (#4752) 1 year ago
gptq_triton.py [inference] add reference and fix some bugs (#4937) 1 year ago
int8_rotary_embedding_kernel.py [inference] Add smmoothquant for llama (#4904) 1 year ago
qkv_matmul_kernel.py [misc] update pre-commit and run all files (#4752) 1 year ago
self_attention_nofusion.py [Refactor] Integrated some lightllm kernels into token-attention (#4946) 1 year ago
smooth_attention.py [inference] add reference and fix some bugs (#4937) 1 year ago
softmax.py [misc] update pre-commit and run all files (#4752) 1 year ago
token_attention_kernel.py [Refactor] Integrated some lightllm kernels into token-attention (#4946) 1 year ago