Aule-Attention : Hardware-agnostic FlashAttention implementation No compilation required. Works on any GPU.
- Dominant language
- Python
- Stars
- 133k
- Forks
- 15.7k
- Avg merge
- 1d 7h
- Merged PRs (30d)
- 158
Description
### Feature Idea
Can we add the aule attetnion at comfy ui is as simple as 3 lines of code.
already has support for comfy ui
ComfyUI / Diffusion Models
'import aule
aule.install() # Patches PyTorch SDPA globally
'
Now all models using F.scaled_dot_product_attention use aule
For diffusion models, causal=False is used automatically
[https://github.com/AuleTechnologies/Aule-Attention](url)
### Existing Solutions
_No response_
### Other
_No response_
Contributor guide
Research direction
The issue names no ComfyUI files or tests; start by locating where PyTorch F.scaled_dot_product_attention is used and reviewing Aule's installation requirements. Done should mean the proposed Aule integration works without compilation and does not break existing diffusion-model attention paths, with GPU validation identified.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- ai
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Quiet
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100