Cuiqing Li (李崔卿) 28052a71fb [Kernels]Update triton kernels into 2.1.0 (#5046) ... * update flash-context-attention * adding kernels * fix * reset * add build script * add building process * add llama2 exmaple * add colossal-llama2 test * clean * fall back test setting * fix test file * clean * clean * clean --------- Co-authored-by: cuiqing.li <lixx336@gmail.com>		2023-11-16 16:43:15 +08:00
..
__init__.py	[Inference] Dynamic Batching Inference, online and offline (#4953)	2023-10-30 10:52:19 +08:00
context_attention.py	[Kernels]Update triton kernels into 2.1.0 (#5046)	2023-11-16 16:43:15 +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
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
int8_rotary_embedding_kernel.py	[inference] Add smmoothquant for llama (#4904)	2023-10-16 11:28:44 +08:00
llama_act_combine_kernel.py	[moe] merge moe into main (#4978)	2023-11-02 02:21:24 +00:00
qkv_matmul_kernel.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
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	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
token_attention_kernel.py	[Kernels]Update triton kernels into 2.1.0 (#5046)	2023-11-16 16:43:15 +08:00