flash: Add non-warp-specialized gemmini flash kernel

2024-11-09 19:08:39 -08:00
parent ac42f2dbba
commit 673e07ed43
3 changed files with 731 additions and 12 deletions
--- a/tests/regression/flash_attention/Makefile
+++ b/tests/regression/flash_attention/Makefile
@@ -2,7 +2,9 @@ PROJECT = flash_attention

 SRCS = main.cpp common.h

-VX_SRCS = kernel.gemmini.cpp
+# VX_SRCS = kernel.cpp
+# VX_SRCS = kernel.gemmini.cpp
+VX_SRCS = kernel.gemmini.nowarpspec.cpp
 VX_INCLUDES = flash_impl.hpp ../sgemm_tcore/sgemm_impl.hpp

 OPTS ?= -n16