wu-arch/kernels
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add flash attention kernel skeleton

Browse Source
This commit is contained in:
Hansung Kim
2024-08-14 20:46:09 -07:00
parent 014f7cd06f
commit 692d028afd
6 changed files with 4514 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/regression/flash_attention/Makefile Normal file
View File

@@ -0,0 +1,12 @@
PROJECT = flash_attention
SRCS = main.cpp common.h
VX_SRCS = kernel.cpp
VX_INCLUDES = ../sgemm_tcore/sgemm_impl.hpp
OPTS ?= -n16
VX_CFLAGS += -I../sgemm_tcore
include ../common.mk