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

Switch header configs to flash

Browse Source
This commit is contained in:
Hansung Kim
2024-11-08 21:56:26 -08:00
parent 4e087a8aab
commit 1e3d476e70
3 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kernel/include/gemmini_mmio.h
View File

@@ -9,9 +9,9 @@
// #define SMEM_SIZE 0x4000
// 64KB
// #define SMEM_SIZE 0x10000
// 128KB
// 128KB (FP16 GEMM)
// #define SMEM_SIZE 0x20000
// 256KB
// 256KB (FlashAttention)
#define SMEM_SIZE 0x40000
#define SMEM_MASK (SMEM_SIZE - 1)