Browse Source

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.
tags/v0.3.18
Wangyang Guo 4 years ago
parent
commit
76ea8db4da
2 changed files with 6 additions and 0 deletions
  1. +3
    -0
      Makefile.system
  2. +3
    -0
      cmake/system.cmake

+ 3
- 0
Makefile.system View File

@@ -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


+ 3
- 0
cmake/system.cmake View File

@@ -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 ()


Loading…
Cancel
Save