vllm/_custom_ops.py · a982f5b5ea4a1932424927ea357b532d0e45caf1 · OpenDAS / vllm_cscc

[kernel][perf] support uncontiguous input for rms_norm kernel (#28103) · a982f5b5

zhrrr authored Nov 21, 2025


Signed-off-by: zhuhaoran <zhuhaoran.zhr@alibaba-inc.com>
Signed-off-by: izhuhaoran <izhuhaoran@qq.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>

a982f5b5

_custom_ops.py 83.4 KB

Replace _custom_ops.py