Add Memory Efficient Attention decode support and tests for ONNX Attention#27851

Closed

titaiwangms wants to merge 8 commits intomainfrom

feature/mea-decode-support

Commits on Mar 25, 2026

Add MEA+decode test cases for ONNX Attention LLM op

titaiwangms
and
Copilot
committed
Add MEA+decode support in ONNX Attention LLM op

titaiwangms
and
Copilot
committed
Fix MEA eligibility: skip decode when head_size != v_head_size

titaiwangms
and
Copilot
committed
Fix review findings: use v_head_size for V ops, add safety comment

titaiwangms
and
Copilot
committed
Fix test review findings for MEA+decode tests

titaiwangms
and
Copilot
committed
Zero present buffers before concat to prevent NaN propagation

titaiwangms
and
Copilot
committed
Add asymmetric head_size regression test for MEA fallback

titaiwangms
and
Copilot
committed
Fix FURB110 lint: use `or` instead of ternary for v_head_size

titaiwangms
and
Copilot
committed