Skip to content

Commit 38d528c

Browse files
author
Kwok Cheung Yeung
committed
Change DGEMM tiling size from 8x8 to 4x8 for the RISC-V ZVL256B architecture
Testing has shown that 4x8 or 4x4 performs better than the original 8x8 tiling size for this kernel. As we do not wish to perturb the behaviour of the DTRMM kernel at this point, the DTRMM tiling size is explicitly set to the original 8x8.
1 parent dbdfea8 commit 38d528c

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

cmake/prebuild.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1475,7 +1475,7 @@ endif ()
14751475
"#define L2_ASSOCIATIVE 4\n")
14761476
set(SGEMM_UNROLL_M 16)
14771477
set(SGEMM_UNROLL_N 8)
1478-
set(DGEMM_UNROLL_M 8)
1478+
set(DGEMM_UNROLL_M 4)
14791479
set(DGEMM_UNROLL_N 8)
14801480
set(CGEMM_UNROLL_M 8)
14811481
set(CGEMM_UNROLL_N 8)

param.h

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3231,7 +3231,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32313231
#define SGEMM_DEFAULT_UNROLL_M 16
32323232
#define SGEMM_DEFAULT_UNROLL_N 8
32333233

3234-
#define DGEMM_DEFAULT_UNROLL_M 8
3234+
#define DGEMM_DEFAULT_UNROLL_M 4
32353235
#define DGEMM_DEFAULT_UNROLL_N 8
32363236

32373237
#define CGEMM_DEFAULT_UNROLL_M 8
@@ -3240,6 +3240,9 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32403240
#define ZGEMM_DEFAULT_UNROLL_M 8
32413241
#define ZGEMM_DEFAULT_UNROLL_N 4
32423242

3243+
#define DTRMM_DEFAULT_UNROLL_M 8
3244+
#define DTRMM_DEFAULT_UNROLL_N 8
3245+
32433246
#undef SHGEMM_DEFAULT_P
32443247
#define SHGEMM_DEFAULT_P 128
32453248
#undef SBGEMM_DEFAULT_P

0 commit comments

Comments
 (0)