yindo/flash-attention
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention.git synced 2026-07-01 21:04:02 -04:00
Code Issues Packages Projects Releases Wiki Activity

doc

Browse Source
This commit is contained in:
Henry Tsang
2026-01-15 20:38:42 -08:00
parent 08e65188b5
commit 94f034800e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flash_attn/cute/mask.py
+1 -1
View File
@@ -90,7 +90,7 @@ def mask_r2p_dual_bound(
right_s = max(col_limit_right - s * 24, 0)
left_s = max(col_limit_left - s * 24, 0)
# XOR creates range mask: bits left_s..(right_s-1) are 1
# bits (right-1)..left are 1
mask_right = (1 << right_s) - 1
mask_left = (1 << left_s) - 1
mask_range = mask_right & ~ mask_left