transformers
Fall back to the AITER kernel for flash_attention_2 on ROCm
#49608
Merged
Overview
Commits
2
Changes
View On GitHub

Fall back to the AITER kernel for flash_attention_2 on ROCm #49608

Abdennacer-Badaoui merged 2 commits into huggingface:main from Abdennacer-Badaoui:rocm-fa2-aiter-fallback
Abdennacer-Badaoui
Abdennacer-Badaoui Fall back to the AITER kernel for flash_attention_2 on ROCm
c65a9081
Abdennacer-Badaoui Abdennacer-Badaoui requested a review from vasqu vasqu 1 day ago
vasqu
vasqu approved these changes on 2026-10-14
HuggingFaceDocBuilderDev
Abdennacer-Badaoui Add a ROCm test for the flash_attention_2 AITER fallback
987d680c
github-actions
Abdennacer-Badaoui Abdennacer-Badaoui merged 2126f65b into main 12 hours ago
Abdennacer-Badaoui Abdennacer-Badaoui deleted the rocm-fa2-aiter-fallback branch 12 hours ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum