tests/neuron/test_prefix_prefill.py · afd0da2186c1d58fb48e138df0a2f548612b5d7d · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[Neuron][Kernel] NKI-based flash-attention kernel with paged KV cache (#11277) · ddee88d0
Liangfu Chen authored Jan 27, 2025
```
Signed-off-by: Liangfu Chen <liangfc@amazon.com>
Co-authored-by: Jiangfei Duan <jfduan@outlook.com>
```
ddee88d0

test_prefix_prefill.py 14.9 KB