Cuiqing Li (李崔卿) bce919708f [Kernels]added flash-decoidng of triton (#5063) ...

* added flash-decoidng of triton based on lightllm kernel

* add req

* clean

* clean

* delete build.sh

---------

Co-authored-by: cuiqing.li <lixx336@gmail.com>

2023-11-20 13:58:29 +08:00

..

__init__.py

[Inference] Dynamic Batching Inference, online and offline (#4953)

2023-10-30 10:52:19 +08:00

context_attention.py

[Kernels]added flash-decoidng of triton (#5063)

2023-11-20 13:58:29 +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

[Kernels]added flash-decoidng of triton (#5063)

2023-11-20 13:58:29 +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