flash: Fix online softmax for DMA layout

2024-09-07 23:21:28 -07:00
parent 2e1485877d
commit c51dc4902d
3 changed files with 54 additions and 48 deletions
--- a/tests/regression/flash_attention/Makefile
+++ b/tests/regression/flash_attention/Makefile
@@ -3,7 +3,7 @@ PROJECT = flash_attention
 SRCS = main.cpp common.h

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

 OPTS ?= -n16