Skip to content

fix(attention): remove stale barrier_O arrive in empty-tile fast path - #13

Merged
fdz-1999 merged 1 commit into
fdz-1999:feature/block-extendfrom
leon062112:fix/block-extend-barrier-parity
Aug 18, 2026
Merged

fdz-1999 merged 1 commit into
fdz-1999:feature/block-extendfrom
leon062112:fix/block-extend-barrier-parity

Conversation

@leon062112

Copy link
Copy Markdown

No description provided.

In BatchPrefillWithKVCacheKernel, when a work tile hits num_kv_tiles <= 0
and exits early via store_zero + continue, the epilogue barrier parity is
now handled internally by store_zero, so the manual barrier_O arrive from
warp NUM_WARPS-1 is no longer needed — and can cause a spurious wake when
the epilogue protocol has already moved on.

Co-Authored-By: Claude <noreply@anthropic.com>
@fdz-1999
fdz-1999 merged commit 5c47e80 into fdz-1999:feature/block-extend Aug 18, 2026
3 of 4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants