feat: enable logsumexp output in fused SDPA dispatch#3594

Closed

adurham wants to merge 3 commits into

ml-explore:mainfrom

adurham:pr/sdpa-logsumexp-output

Commits on May 27, 2026

feat: add output_logsumexp function constant to steel_attention kernel

Thump604
authored and
Adam Durham
committed
feat: enable logsumexp output in fused SDPA dispatch

Thump604
authored and
Adam Durham
committed
test: add logsumexp output correctness tests for fused SDPA

Thump604
authored and
Adam Durham
committed