|
- if(CMAKE_TOOLCHAIN_FILE)
- set(LIBRARY_OUTPUT_PATH_ROOT ${CMAKE_BINARY_DIR} CACHE PATH "root for library output, set this to change where android libs are compiled to")
- # get absolute path, but get_filename_component ABSOLUTE only refer with source dir, so find_file here :(
- get_filename_component(CMAKE_TOOLCHAIN_FILE_NAME ${CMAKE_TOOLCHAIN_FILE} NAME)
- find_file(CMAKE_TOOLCHAIN_FILE ${CMAKE_TOOLCHAIN_FILE_NAME} PATHS ${CMAKE_SOURCE_DIR} NO_DEFAULT_PATH)
- message(STATUS "CMAKE_TOOLCHAIN_FILE = ${CMAKE_TOOLCHAIN_FILE}")
- endif()
-
- if(NOT DEFINED CMAKE_INSTALL_PREFIX)
- set(CMAKE_INSTALL_PREFIX "${CMAKE_BINARY_DIR}/install" CACHE PATH "Installation Directory")
- endif()
- message(STATUS "CMAKE_INSTALL_PREFIX = ${CMAKE_INSTALL_PREFIX}")
-
- # This is the minimum cmake version supporting the modern cmake OpenMP module below
- # modern cmake requires 3.0 anyway and android cmake requires 3.6
- cmake_minimum_required(VERSION 3.1.3)
-
- if(NOT CMAKE_BUILD_TYPE)
- set(CMAKE_BUILD_TYPE release CACHE STRING "Choose the type of build" FORCE)
- endif()
-
- project(ncnn)
-
- option(NCNN_OPENMP "openmp support" ON)
- option(NCNN_STDIO "load model from external file" ON)
- option(NCNN_STRING "plain and verbose string" ON)
- option(NCNN_INSTALL_SDK "Install ncnn library and headers" ON)
- option(NCNN_OPENCV "minimal opencv structure emulation" OFF)
- option(NCNN_BENCHMARK "print benchmark information for every layer" OFF)
- option(NCNN_PIXEL "convert and resize from/to image pixel" ON)
- option(NCNN_PIXEL_ROTATE "rotate image pixel orientation" OFF)
- option(NCNN_CMAKE_VERBOSE "print verbose cmake messages" OFF)
- option(NCNN_VULKAN "vulkan compute support" OFF)
- option(NCNN_REQUANT "auto merge int8 quant and dequant" OFF)
- option(NCNN_IM2COL_SGEMM "im2col sgemm support" OFF)
-
- if(NCNN_OPENMP)
- find_package(OpenMP)
- # For CMake < 3.9, we need to make the target ourselves
- if(NOT TARGET OpenMP::OpenMP_CXX)
- find_package(Threads REQUIRED)
- add_library(OpenMP::OpenMP_CXX IMPORTED INTERFACE)
- set_property(TARGET OpenMP::OpenMP_CXX
- PROPERTY INTERFACE_COMPILE_OPTIONS ${OpenMP_CXX_FLAGS})
- # Only works if the same flag is passed to the linker; use CMake 3.9+ otherwise (Intel, AppleClang)
- set_property(TARGET OpenMP::OpenMP_CXX
- PROPERTY INTERFACE_LINK_LIBRARIES ${OpenMP_CXX_FLAGS} Threads::Threads)
- endif()
- endif()
-
- if(NCNN_VULKAN)
- find_package(Vulkan REQUIRED)
- include_directories(${Vulkan_INCLUDE_DIR})
- endif()
-
- if(WIN32)
- add_definitions(-D_SCL_SECURE_NO_WARNINGS -D_CRT_SECURE_NO_DEPRECATE)
- add_definitions(-DNOMINMAX)
- else()
- add_definitions(-Wall -Wextra -Wno-unused-function)
-
- add_definitions(-fPIC)
- if(CMAKE_BUILD_TYPE MATCHES "(Release|RELEASE|release)")
- add_definitions(-Ofast)
-
- add_definitions(-ffast-math)
- endif()
-
- if(PI3)
- add_definitions(-march=native)
- add_definitions(-mfpu=neon)
- add_definitions(-mfloat-abi=hard)
- add_definitions(-D__ARM_NEON)
- add_definitions(-D__ANDROID__)
- endif()
- # add_definitions(-march=native)
- # add_definitions(-flto)
- add_definitions(-fvisibility=hidden -fvisibility-inlines-hidden)
- endif()
-
- if(ANDROID)
- # disable shared library on android
- set_property(GLOBAL PROPERTY TARGET_SUPPORTS_SHARED_LIBS FALSE)
- set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fno-rtti -fno-exceptions")
- elseif(IOS)
- # disable shared library on xcode ios
- set_property(GLOBAL PROPERTY TARGET_SUPPORTS_SHARED_LIBS FALSE)
- set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fno-rtti -fno-exceptions")
- endif()
-
- ##############################################
-
- # add_subdirectory(examples)
- add_subdirectory(benchmark)
- add_subdirectory(src)
- if(NOT ANDROID AND NOT IOS)
- add_subdirectory(tools)
- endif()
|