tests/neuron/test_prefix_prefill.py · ec5e299c4d99a0c0aeeaa63979d7f95312211d95 · 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