Skip to content

Pull requests: Dao-AILab/flash-attention

New pull request New
110 Open 438 Closed
110 Open 438 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[WIP] support the flash api for Ascend
#2246 opened Feb 10, 2026 by AnyFree813 Loading…
[AMD] Adding gfx1150/51 for AMD RDNA arch
#2245 opened Feb 10, 2026 by saeid-rostami • Draft
Add gfx1150/gfx1151 (RDNA 3.5) to RDNA_ARCHS
#2243 opened Feb 9, 2026 by rwfsmith Loading…
2
[Cute,Flex,Fwd] Allow vectorized score_mod definitions
#2236 opened Feb 5, 2026 by reubenconducts Loading…
7
[AMD] Migrate to Triton Backend to Aiter
#2230 opened Feb 4, 2026 by micmelesse Loading…
1
Nicer headdim error message
#2227 opened Feb 4, 2026 by drisspg Loading…
8
[WIP] varlen blocksparsity
#2224 opened Feb 2, 2026 by reubenconducts • Draft
[Draft][Cute,Fwd,Sm120] FA Cute DSL sm12x
#2222 opened Feb 2, 2026 by johnnynunez • Draft
1
[Ai-assisted] CLC work stealing
#2218 opened Jan 31, 2026 by drisspg Loading…
11
[ROCM] Add support with Infinity Cache (LLC) awareness for performance improvement - [PR#2147 rebased on PR#2178]
#2217 opened Jan 29, 2026 by tianwyan Loading…
16
Add shift scheduler for deterministic full‑mask FA3 bwd on Hopper (sm90)
#2207 opened Jan 23, 2026 by tie-pilot-qxw Loading…
3
Add loc info & Fix api changes for CuTeDSL 4.4
#2204 opened Jan 23, 2026 by keithzzzzz Loading…
2
BWD sm100 2cta
#2202 opened Jan 23, 2026 by tzadouri Loading…
5
[Cute, SM100] Fix comment in tmem_p_offset
#2201 opened Jan 22, 2026 by Edenzzzz Loading…
Warn when ninja is missing
#2191 opened Jan 17, 2026 by blueberrycongee Loading…
1
Fix compute_block_sparsity import in benchmark_mask_mod
#2190 opened Jan 17, 2026 by blueberrycongee Loading…
1
[Cute][Testing] Protyping a fast test mode for Cute
#2188 opened Jan 16, 2026 by drisspg Loading…
1
[Cute,Fwd,Sm100] support irregular qhead / kvhead ratios
#2186 opened Jan 16, 2026 by timmy-feng • Draft
[Cute] Add torch.compile support for FA4
#2164 opened Jan 9, 2026 by gilfordting Loading…
8
Update mha_fwd.cpp, Normalize the commented-out parameters
#2160 opened Jan 9, 2026 by breakfei Loading…
Update schema in test_flash3_bw_compatibility
#2153 opened Jan 8, 2026 by guilhermeleobas • Draft
6
Add FLASH_ATTENTION_FORCE_NON_STABLE_API option to allow building on NVidia Pytorch 25.09 image
#2140 opened Jan 5, 2026 by jp-gr Loading…
2
[ROCM] Fix AMD Triton backend crash when dropout != 0 and return_attn_probs = False
#2111 opened Dec 30, 2025 by Logiquo Loading…
1
1
[Cute,Fwd,Sm100] fp8 e4m3 and e5m2 support
#2109 opened Dec 29, 2025 by dcw02 Loading…
4
refactor llama test
#2107 opened Dec 29, 2025 by m3ngyang Loading…
ProTip! Exclude everything labeled bug with -label:bug.