From 76ea8db4da1a651bb4de744162de1ecfc6762e7c Mon Sep 17 00:00:00 2001 From: Wangyang Guo Date: Thu, 5 Aug 2021 02:57:58 +0000 Subject: [PATCH] Small Matrix: enable by default for x86_64 arch If no customized GEMM_SMALL_M_PERMIT kernel defined, it will just by pass to normal path. --- Makefile.system | 3 +++ cmake/system.cmake | 3 +++ 2 files changed, 6 insertions(+) diff --git a/Makefile.system b/Makefile.system index 20d8d2f..20db80d 100644 --- a/Makefile.system +++ b/Makefile.system @@ -245,6 +245,9 @@ ONLY_CBLAS = 0 endif #For small matrix optimization +ifeq ($(ARCH), x86_64) +SMALL_MATRIX_OPT = 1 +endif ifeq ($(SMALL_MATRIX_OPT), 1) CCOMMON_OPT += -DSMALL_MATRIX_OPT endif diff --git a/cmake/system.cmake b/cmake/system.cmake index e51dc1f..7d26729 100644 --- a/cmake/system.cmake +++ b/cmake/system.cmake @@ -258,6 +258,9 @@ if (NEED_PIC) endif() endif () +if (X86_64) + set(SMALL_MATRIX_OPT TRUE) +endif () if (SMALL_MATRIX_OPT) set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT") endif () -- 2.7.4