This website requires JavaScript.
Explore
Help
Sign In
wu-arch
/
kernels
Watch
1
Star
0
Fork
0
You've already forked kernels
Code
Issues
Pull Requests
Actions
Packages
Projects
Releases
Wiki
Activity
Files
53dfc690b9b05da4c27dc2699c5d28ffed9cc4c8
kernels
/
tests
/
regression
/
flash_attention
History
Hansung Kim
53dfc690b9
flash: Allocate smem properly for rowsum and scratch
2024-08-14 21:50:20 -07:00
..
common.h
Add flash attention kernel skeleton
2024-08-14 20:46:09 -07:00
flash_attention
Add flash attention kernel skeleton
2024-08-14 20:46:09 -07:00
half.hpp
Add flash attention kernel skeleton
2024-08-14 20:46:09 -07:00
kernel.cpp
flash: Allocate smem properly for rowsum and scratch
2024-08-14 21:50:20 -07:00
main.cpp
Add flash attention kernel skeleton
2024-08-14 20:46:09 -07:00
Makefile
Add flash attention kernel skeleton
2024-08-14 20:46:09 -07:00
