yuehuayingxueluo 86b63f720c [Inference]Adapted to the triton attn kernels (#5264) ...

* adapted to the triton attn kernels

* fix pad input

* adapted to copy_kv_to_blocked_cache

* fix ci test

* update kv memcpy

* remove print

2024-01-17 16:03:10 +08:00

..

benchmark_llama.py

[Inference]Adapted to the triton attn kernels (#5264)

2024-01-17 16:03:10 +08:00

build_smoothquant_weight.py

[inference] refactor examples and fix schedule (#5077)

2023-11-21 10:46:03 +08:00

run_benchmark.sh

[Inference]Adapted to the triton attn kernels (#5264)

2024-01-17 16:03:10 +08:00

run_llama_inference.py

[inference] refactor examples and fix schedule (#5077)

2023-11-21 10:46:03 +08:00