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

flash: Add Gemmini-accelerated kernel

Browse Source
This commit is contained in:
Hansung Kim
2024-09-07 22:40:50 -07:00
parent b3be271b88
commit 2e1485877d
4 changed files with 689 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/regression/flash_attention/Makefile
View File

@@ -2,7 +2,7 @@ PROJECT = flash_attention
SRCS = main.cpp common.h
VX_SRCS = kernel.cpp
VX_SRCS = kernel.gemmini.cpp
VX_INCLUDES = ../sgemm_tcore/sgemm_impl.hpp
OPTS ?= -n16