csrc/cache_kernels.cu · 436e523bf1616bea22a1df78d93d7522311dccc8 · norm / vllm · GitLab

Find file Blame History Permalink

Memcpy kernel for flash attention (#29) · e3cec88a
Siyuan (Ryans) Zhuang authored Apr 10, 2023
```
* optimize

* add benchmark

* add assert

* add test
```
e3cec88a

cache_kernels.cu 14 KB