Repository navigation

Fix int16 overflow in SDPA NAX mask indexing for KV sequences > 32K - #3361

Merged

angeloskath merged 5 commits into

ml-explore:mainfrom

Clydingus:fix/sdpa-nax-int16-overflow

Apr 10, 2026

Merged

Fix int16 overflow in SDPA NAX mask indexing for KV sequences > 32K#3361
angeloskath merged 5 commits into
ml-explore:mainfrom
Clydingus:fix/sdpa-nax-int16-overflow

Commits on Apr 3, 2026
- Fix int16 overflow in SDPA NAX mask indexing for KV sequences > 32K
  Clydingus
  committed
  ccfad15
Commits on Apr 4, 2026
- Add SDPA test for overflow in mask col_pos of KV sequences > 32K
  Clydingus
  committed
  fe183de
- Add unittest skipif not on gpu
  Clydingus
  committed
  6f14a8c
Commits on Apr 9, 2026
- Improve the mask check for NAX attention
  angeloskath
  committed
  ed93902
- Fix condition
  angeloskath
  committed
  6f95bb0