Skip to content

[BugFix] Fix batch_size derivation and relax shape checks in SM90 flash_mask_attn #7210

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
xiaoxiaohehe001 wants to merge 2 commits into
base: develop
Choose a base branch
from
+3 −3

[BugFix] fix_flash_mask_attn_sm90

ec2a545
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
Open

[BugFix] Fix batch_size derivation and relax shape checks in SM90 flash_mask_attn #7210

[BugFix] fix_flash_mask_attn_sm90
ec2a545
Select commit
Loading
Failed to load commit list.
Remove skip-ci labels on new commits
succeeded Apr 7, 2026 in 4s