Skip to content

Add SM120 shared memory guards for tile selection - #13

Merged
ivandobskygithub merged 2 commits into
mainfrom
codex/fix-flashattention-shared-memory-error-wuusn2
Nov 24, 2025
Merged

ivandobskygithub merged 2 commits into
mainfrom
codex/fix-flashattention-shared-memory-error-wuusn2

Conversation

@ivandobskygithub

Copy link
Copy Markdown
Owner

Summary

  • add explicit shared-memory estimator and clamping to the SM90 tile selector to respect the SM120 limit
  • compile a small bridge for tile_size_fwd_sm90 and exercise head/value/paging permutations under the SMEM cap

Testing

  • pytest tests/hopper/test_tile_size_shared_memory.py -q

Codex Task

@ivandobskygithub
ivandobskygithub merged commit 698bf86 into main Nov 24, 2025
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant