vllm/_custom_ops.py · e73ff24e31d29b095e24e160d3b0651d2436d977 · OpenDAS / vllm_cscc

[ROCM][KERNEL] Paged attention for V1 (#15720) · e73ff24e

Aleksandr Malyshev authored Apr 02, 2025


Signed-off-by: Aleksandr Malyshev <maleksan@amd.com>
Signed-off-by: root <root@banff-cyxtera-s65-4.amd.com>
Co-authored-by: Aleksandr Malyshev <maleksan@amd.com>
Co-authored-by: root <root@banff-cyxtera-s65-4.amd.com>

e73ff24e

_custom_ops.py 57.6 KB

Replace _custom_ops.py