WIP
A tiny flash attention implement in python, rust, cuda and c for learning purpose.
- python version
- triton version
- [c version]
- [rust version]
forked from 66RING/tiny-flash-attention
-
Notifications
You must be signed in to change notification settings - Fork 0
Amanda-Barbara/cutlass-flash-attention
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
flash attention tutorial written in python, triton, cuda, cutlass
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Cuda 60.7%
- Python 21.0%
- C++ 17.1%
- Other 1.2%