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

sgemm_gemmini_dma: Fix tile size to (128,64,128)

Browse Source
This commit is contained in:
Hansung Kim
2024-09-10 18:29:26 -07:00
parent 2152c80ffd
commit 28b2eaec8f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/regression/sgemm_gemmini_dma/kernel.cpp
View File

@@ -8,9 +8,9 @@
// fp16 16x16 // fp16 16x16
#define TILE_M 128 #define TILE_M 128
#define TILE_N 128 #define TILE_N 64
#define TILE_K 128 #define TILE_K 128
#define BOUND_INST 0x800080008ULL #define BOUND_INST 0x800040008ULL
#define NUM_THREADS_IN_CLUSTER 512 #define NUM_THREADS_IN_CLUSTER 512
// fp32 8x8 // fp32 8x8
@@ -195,4 +195,4 @@ int main() {
vx_spawn_tasks_contiguous(grid_size, (vx_spawn_tasks_cb)kernel_body, arg); vx_spawn_tasks_contiguous(grid_size, (vx_spawn_tasks_cb)kernel_body, arg);
#endif #endif
return 0; return 0;
} }