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.
This commit is contained in:
Wangyang Guo 2021-08-05 02:57:58 +00:00
parent aa50185647
commit 76ea8db4da
2 changed files with 6 additions and 0 deletions

View File

@ -245,6 +245,9 @@ ONLY_CBLAS = 0
endif endif
#For small matrix optimization #For small matrix optimization
ifeq ($(ARCH), x86_64)
SMALL_MATRIX_OPT = 1
endif
ifeq ($(SMALL_MATRIX_OPT), 1) ifeq ($(SMALL_MATRIX_OPT), 1)
CCOMMON_OPT += -DSMALL_MATRIX_OPT CCOMMON_OPT += -DSMALL_MATRIX_OPT
endif endif

View File

@ -258,6 +258,9 @@ if (NEED_PIC)
endif() endif()
endif () endif ()
if (X86_64)
set(SMALL_MATRIX_OPT TRUE)
endif ()
if (SMALL_MATRIX_OPT) if (SMALL_MATRIX_OPT)
set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT") set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT")
endif () endif ()