Skip to content

[Fix][DLight] Decline GEMV reductions with only batch spatial axes - #20985

Merged
tlopex merged 1 commit into
apache:mainfrom
JCNTH:codex/gemv-batch-reduction-guard
Nov 15, 2026
Merged

tlopex merged 1 commit into
apache:mainfrom
JCNTH:codex/gemv-batch-reduction-guard

Conversation

@JCNTH

@JCNTH JCNTH commented Nov 15, 2026

Copy link
Copy Markdown
Contributor

Weighted centered reductions can look like GEMV when a broadcast operand uses every spatial axis, leaving only batch axes and triggering an assertion; decline those blocks so the reduction schedule handles them instead.

The regression verifies rejection and fallback, all 15 GEMV tests pass using the current pass wrapper with the retained compiler, and the patched compiler builds ECAPA for WebGPU with eight HF parity cases passing in Chrome.

@tlopex
tlopex merged commit 1bc0ce8 into apache:main Nov 15, 2026
8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants