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

Change gemmini_mmio.h to fp16 GEMM setting

Browse Source
This commit is contained in:
Hansung Kim
2025-01-28 16:36:55 -08:00
parent e4c0bbd039
commit d98a414765
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kernel/include/gemmini_mmio.h
View File

@@ -10,9 +10,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)