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

Merge branch 'kernels-flash' into new-cisc

Browse Source
This commit is contained in:
Hansung Kim
2024-11-09 14:42:46 -08:00
parent c114a7a4ab 365b1d8e67
commit dc89309ad0
6 changed files with 170 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
kernel/include/gemmini_mmio.h
View File

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