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

sgemm_tcore: Skip load at last k-iter; do DMA by default

Browse Source
This commit is contained in:
Hansung Kim
2024-07-19 16:37:51 -07:00
parent 1f844fa9e9
commit bca53a9c76
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/regression/sgemm_tcore/Makefile
View File

@@ -2,7 +2,7 @@ PROJECT = sgemm_tcore
SRCS = main.cpp common.h SRCS = main.cpp common.h
VX_SRCS = kernel.activation.cpp VX_SRCS = kernel.cpp
OPTS ?= -n16 OPTS ?= -n16

4
tests/regression/sgemm_tcore/kernel.cpp
View File

@@ -7,7 +7,7 @@
#include "include/gemmini.h" #include "include/gemmini.h"
#include "gemmini_mmio.h" #include "gemmini_mmio.h"
#define GEMMINI_DMA 0 #define GEMMINI_DMA 1
#if SMEM_SIZE == 0x4000 #if SMEM_SIZE == 0x4000
#define SMEM_ADDR_Q0 ((float * const) 0xff000000) #define SMEM_ADDR_Q0 ((float * const) 0xff000000)
#define SMEM_ADDR_Q1 ((float * const) 0xff001000) #define SMEM_ADDR_Q1 ((float * const) 0xff001000)
@@ -368,7 +368,7 @@ inline void thread_block_gemm(kernel_arg_t *__UNIFORM__ arg,
// this is either done using DMA or SIMT cores depending on GEMMINI_DMA // this is either done using DMA or SIMT cores depending on GEMMINI_DMA
#if (GEMMINI_DMA == 1) #if (GEMMINI_DMA == 1)
if (tid_in_threadblock == 0) { if ((tid_in_threadblock == 0) && ((block_k * BK) != (dim_k - BK))) {
// configure dma gmem address to load from // configure dma gmem address to load from
// FIXME: block_k is wrong // FIXME: block_k is wrong
ROCC_INSTRUCTION_RS1_RS2( ROCC_INSTRUCTION_RS1_RS2(