vllm/_custom_ops.py · 6cc6d92be5e64f449e31e06d43c80955f646809e · OpenDAS / vllm_cscc

[CI][AMD][BugFix] Update wvSplitK (and other skinny_gemm wrappers) to ensure... · 6cc6d92b

rasmith authored Jan 23, 2026


[CI][AMD][BugFix] Update wvSplitK (and other skinny_gemm wrappers) to ensure tensors passed will be made contiguous for the kernel (#32831)
Signed-off-by: Randall Smith <ransmith@amd.com>
Co-authored-by: Randall Smith <ransmith@amd.com>

6cc6d92b

_custom_ops.py 94.9 KB

Replace _custom_ops.py