Use lower-right causal mask alignment consistently - #2967

Merged

awni merged 10 commits into

ml-explore:mainfrom

Anri-Lombard:fix/sdpa-causal-mask-offset

Jan 29, 2026

Commits on Jan 4, 2026

Document causal mask alignment in scaled_dot_product_attention
Anri-Lombard
committed
c55b3f5

Commits on Jan 19, 2026

Fix causal mask documentation to describe conditional alignment behavior
Anri-Lombard
committed
f434231

Commits on Jan 23, 2026

Use bottom-right causal mask alignment consistently
Anri-Lombard
committed
ec68023
Support qL > kL with causal mask using lower-right alignment
Anri-Lombard
committed
431ccbf
Update docstring to reflect consistent lower-right causal alignment
Anri-Lombard
committed
4bc3b75
Merge remote-tracking branch 'upstream/main' into fix/sdpa-causal-mask-offset
Anri-Lombard
committed
fc834bf
Remove zero-row handling for qL > kL causal case
Anri-Lombard
committed
07ba3f5

Commits on Jan 26, 2026

Update mlx/fast.cpp

Anri-Lombard
and
awni
authored
1776be7

Commits on Jan 27, 2026

Revert fast path for qL > kL causal case
Anri-Lombard
committed
49f19fc

Commits on Jan 28, 2026

Fix test slicing for transpose=True case
Anri-Lombard
committed
132121f