tests/neuron/test_prefix_prefill.py · 71face8540042e309b01a18e5f184cc700a8bc6d · OpenDAS / vllm_cscc

[Neuron][Kernel] Support Longer Sequences in NKI-based Flash PagedAttention... · e92694b6

Lingfan Yu authored Feb 11, 2025


[Neuron][Kernel] Support Longer Sequences in NKI-based Flash PagedAttention and Improve Efficiency (#12921)
Signed-off-by: Lingfan Yu <lingfany@amazon.com>

e92694b6

test_prefix_prefill.py 15.3 KB

Replace test_prefix_prefill.py