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