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

Increase SMEM size for flash

Browse Source
This commit is contained in:
Hansung Kim
2025-01-30 01:17:38 -08:00
parent efd2d232fe
commit 4f12227327
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/include/gemmini_mmio.h
View File

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