vllm
[CI][AMD][BugFix] Update wvSplitK (and other skinny_gemm wrappers) to ensure tensors passed will be made contiguous for the kernel
#32831
Merged

[CI][AMD][BugFix] Update wvSplitK (and other skinny_gemm wrappers) to ensure tensors passed will be made contiguous for the kernel #32831

rasmith
update wvSplitK custom op to ensure contiguous tensors
ea8b0957
mergify mergify added rocm
mergify mergify added bug
gemini-code-assist
gemini-code-assist commented on 2026-01-22
typo
98a02d26
update warning message to note performance loss to user
f060711a
rasmith Merge branch 'main' into rasmith_fix_wvsplitk
e92c9f4d
update the other skinny gemm calls
fd09296e
add comment for TODO to skinny_gemms.cu
17ee0642
rasmith rasmith requested a review from gshtras gshtras 192 days ago
rasmith rasmith requested a review from tjtanaa tjtanaa 192 days ago
pass fn name as string
4e2995a6
rasmith rasmith changed the title [CI][AMD][BugFix] Update _custom_ops.wvSplitK to ensure tensors passed will be made contiguous for the kernel [CI][AMD][BugFix] Update wvSplitK (and other skinny_gemm wrappers) to ensure tensors passed will be made contiguous for the kernel 192 days ago
update LLM1
c34bd6b2
tjtanaa
tjtanaa commented on 2026-01-22
mawong-amd
mawong-amd commented on 2026-01-22
gshtras
gshtras commented on 2026-01-22
fix typo
e8407279
uncomment code
44c8e9e6
skinny gemms rocm only, so remove if current_platform.is_rocm
013c38fc
remove fn name arg
b0c8fd4b
remove typo
ce3042a1
tjtanaa
tjtanaa approved these changes on 2026-01-23
tjtanaa Merge branch 'main' into rasmith_fix_wvsplitk
f64b4729
tjtanaa tjtanaa enabled auto-merge (squash) 190 days ago
tjtanaa tjtanaa added ready
tjtanaa tjtanaa merged 6cc6d92b into main 190 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone