Cuiqing Li 3a41e8304e [Refactor] Integrated some lightllm kernels into token-attention (#4946) ... * add some req for inference * clean codes * add codes * add some lightllm deps * clean codes * hello * delete rms files * add some comments * add comments * add doc * add lightllm deps * add lightllm cahtglm2 kernels * add lightllm cahtglm2 kernels * replace rotary embedding with lightllm kernel * add some commnets * add some comments * add some comments * add * replace fwd kernel att1 * fix a arg * add * add * fix token attention * add some comments * clean codes * modify comments * fix readme * fix bug * fix bug --------- Co-authored-by: cuiqing.li <lixx336@gmail.com> Co-authored-by: CjhHa1 <cjh18671720497@outlook.com>		2023-10-19 22:22:47 +08:00
..
kernel_utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_bloom_context_attention.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_copy_kv_dest.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_layernorm_triton.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_llama_context_attention.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_self_attention_nonfusion.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_softmax.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_token_attn_fwd.py	[Refactor] Integrated some lightllm kernels into token-attention (#4946)	2023-10-19 22:22:47 +08:00
test_token_softmax.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00