Liangfu Chen ddee88d0ff [Neuron][Kernel] NKI-based flash-attention kernel with paged KV cache (#11277) ...

Signed-off-by: Liangfu Chen <liangfc@amazon.com>
Co-authored-by: Jiangfei Duan <jfduan@outlook.com>

2025-01-27 17:31:16 -08:00

..

test_prefix_prefill.py

[Neuron][Kernel] NKI-based flash-attention kernel with paged KV cache (#11277)

2025-01-27 17:31:16 -08:00