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

volta change

Browse Source
This commit is contained in:
Richard Yan
2025-01-29 22:16:39 -08:00
parent 91a82c9f0f
commit 6e279c905f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kernels/sgemm_tcore/sgemm_impl.hpp
View File

@@ -110,7 +110,7 @@ static_assert(WMITER * WNITER * TCM * TCN * NUM_WARPS * CORES_PER_CLUSTER ==
// result matrix will be stored in a swizzled form in the global memory. // result matrix will be stored in a swizzled form in the global memory.
#define WMMA_STORE_FAST 0 #define WMMA_STORE_FAST 0
#define GEMMINI_DMA 1 #define GEMMINI_DMA 0
#define GEMMINI_DMA_FAST 1 #define GEMMINI_DMA_FAST 1
#if SMEM_SIZE == 0x4000 #if SMEM_SIZE == 0x4000
#define SMEM_ADDR_Q0 ((float * const) 0xff000000) #define SMEM_ADDR_Q0 ((float * const) 0xff000000)