yuehuayingxueluo 600881a8ea [Inference]Add CUDA KVCache Kernel (#5406) ...

* add cuda KVCache kernel

* annotation benchmark_kvcache_copy

* add use cuda

* fix import path

* move benchmark scripts to example/

* rm benchmark codes in test_kv_cache_memcpy.py

* rm redundancy codes

* rm redundancy codes

* pr was modified according to the review

2024-02-28 14:36:50 +08:00

..

benchmark_ops

[Inference]Add CUDA KVCache Kernel (#5406)

2024-02-28 14:36:50 +08:00

benchmark_llama.py

Optimized the execution interval time between cuda kernels caused by view and memcopy (#5390)

2024-02-21 13:23:57 +08:00

build_smoothquant_weight.py

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

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

run_benchmark.sh

[Fix/Inference] Fix format of input prompts and input model in inference engine (#5395)

2024-02-23 10:51:35 +08:00

run_llama_inference.py

[npu] change device to accelerator api (#5239)

2024-01-09 10:20:05 +08:00