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/flash_impl.hpp
View File

@@ -10,7 +10,7 @@
constexpr uint32_t ROWMAX_SETS = 3;
constexpr bool DEBUG = true;
constexpr bool WARP_SPECIALIZED = true;
constexpr bool WARP_SPECIALIZED = false;
constexpr uint32_t DEV_FAKE_SMEM_START_ADDR = 0xf0000000;