commit b36582b974aa01e440d1734ca1767f1d33aa682d
author Marat Dukhan <maratek@google.com> Tue Dec 08 11:16:28 2020 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Dec 08 11:17:14 2020 -0800
tree d1907592c29ea3a8f56a76bbdbe6d06e04e510f4
parent cfbed0a72f0000eb704eccfb708ab91876fd0ea4

Enable sparse inference by default

PiperOrigin-RevId: 346367914

CMakeLists.txt

diff --git a/CMakeLists.txt b/CMakeLists.txt
index d975e2a..9436b12 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -16,6 +16,7 @@
 SET_PROPERTY(CACHE XNNPACK_LIBRARY_TYPE PROPERTY STRINGS default static shared)
 OPTION(XNNPACK_ENABLE_ASSEMBLY "Build XNNPACK with assembly micro-kernels" ON)
 OPTION(XNNPACK_ENABLE_MEMOPT "Build XNNPACK with optimized memory allocation scheme" ON)
+OPTION(XNNPACK_ENABLE_SPARSE "Build XNNPACK with graph rewriting for sparse inference" ON)
 OPTION(XNNPACK_BUILD_TESTS "Build XNNPACK unit tests" ON)
 OPTION(XNNPACK_BUILD_BENCHMARKS "Build XNNPACK benchmarks" ON)
 OPTION(XNNPACK_USE_SYSTEM_LIBS "Use system-provided dependency libraries" OFF)
@@ -39,6 +40,12 @@
   ADD_DEFINITIONS(-DXNN_ENABLE_MEMOPT=0)
 ENDIF()
 
+IF(XNNPACK_ENABLE_SPARSE)
+  ADD_DEFINITIONS(-DXNN_ENABLE_SPARSE=1)
+ELSE()
+  ADD_DEFINITIONS(-DXNN_ENABLE_SPARSE=0)
+ENDIF()
+
 IF(CMAKE_C_COMPILER_ID STREQUAL "MSVC")
   # Disable "unary minus operator applied to unsigned type, result still unsigned" warning
   ADD_COMPILE_OPTIONS("/wd4146")