# ****************************************************************************** # Copyright 2017-2020 Intel Corporation # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # ****************************************************************************** cmake_minimum_required (VERSION 3.4) # set directory where the custom finders live set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/cmake/Modules/") if("${CMAKE_SOURCE_DIR}" STREQUAL "${CMAKE_BINARY_DIR}") message(FATAL_ERROR "In-source builds are not allowed.") endif() if (CMAKE_BUILD_TYPE) set(RELEASE_TYPES Debug Release RelWithDebInfo MinSizeRel) list(FIND RELEASE_TYPES ${CMAKE_BUILD_TYPE} INDEX_FOUND) if (${INDEX_FOUND} EQUAL -1) message(FATAL_ERROR "CMAKE_BUILD_TYPE must be one of Debug, Release, RelWithDebInfo, or MinSizeRel") endif() endif() include(git_tags) include(var_functions) NGRAPH_GET_VERSION_LABEL() string(REGEX MATCH "([0-9?]+)\\.([0-9?]+)\\.([0-9?]+)(-(rc|dev)\\.[0-9?]+)?" NGRAPH_VERSION_SHORT "${NGRAPH_VERSION_LABEL}") string(REGEX REPLACE "-rc." "rc" NGRAPH_WHEEL_VERSION "${NGRAPH_VERSION_SHORT}") string(REGEX REPLACE "-dev." "dev" NGRAPH_WHEEL_VERSION "${NGRAPH_WHEEL_VERSION}") string(REGEX MATCH "([0-9?]+)\\.([0-9?]+)" NGRAPH_API_VERSION "${NGRAPH_VERSION_LABEL}") string(REGEX MATCH "[^v](.*)" NGRAPH_VERSION "${NGRAPH_VERSION_LABEL}") string(REPLACE "." ";" NGRAPH_VERSION_PARTS "${NGRAPH_VERSION_SHORT}") list(GET NGRAPH_VERSION_PARTS 0 NGRAPH_VERSION_MAJOR) list(GET NGRAPH_VERSION_PARTS 1 NGRAPH_VERSION_MINOR) list(GET NGRAPH_VERSION_PARTS 2 NGRAPH_VERSION_PATCH) configure_file(VERSION.in VERSION) message(STATUS "NGRAPH_VERSION ${NGRAPH_VERSION}") message(STATUS "NGRAPH_VERSION_SHORT ${NGRAPH_VERSION_SHORT}") message(STATUS "NGRAPH_WHEEL_VERSION ${NGRAPH_WHEEL_VERSION}") message(STATUS "NGRAPH_API_VERSION ${NGRAPH_API_VERSION}") set(NGRAPH_INCLUDE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/src ) if (APPLE) # Enable MACOS_RPATH by default. cmake_policy(SET CMP0042 NEW) # Enable CMAKE_<LANG>_COMPILER_ID AppleClang cmake_policy(SET CMP0025 NEW) endif() project (ngraph) if (UNIX AND NOT APPLE) set(LINUX TRUE) endif() # APPLE: Set CMAKE_OSX_SYSROOT if not set already. if (APPLE) execute_process(COMMAND sw_vers -productVersion OUTPUT_VARIABLE OSX_FULL_VERSION OUTPUT_STRIP_TRAILING_WHITESPACE) string(REGEX REPLACE "^([0-9]+\\.[0-9]+).*$" "\\1" OSX_SHORT_VERSION "${OSX_FULL_VERSION}") message(STATUS "Detected: OSX ${OSX_SHORT_VERSION}") if (CMAKE_OSX_SYSROOT) message(STATUS "Using CMAKE_OSX_SYSROOT: ${CMAKE_OSX_SYSROOT}") else() execute_process(COMMAND xcode-select -p OUTPUT_VARIABLE APPLE_DEV_DIR OUTPUT_STRIP_TRAILING_WHITESPACE) if("${APPLE_DEV_DIR}" STREQUAL "/Library/Developer/CommandLineTools") # Command line tools only set(XCODE_ISYSROOT ${APPLE_DEV_DIR}/SDKs/MacOSX.sdk) message(STATUS "Trying command line tool sdk at ${XCODE_ISYSROOT}.") if(NOT EXISTS ${XCODE_ISYSROOT}) message(FATAL_ERROR "Cannot find macos sdk.") endif() else() # Xcode is installed set(XCODE_ISYSROOT ${APPLE_DEV_DIR}/Platforms/MacOSX.platform/Developer/SDKs/MacOSX${OSX_SHORT_VERSION}.sdk) message(STATUS "Trying Xcode sdk at ${XCODE_ISYSROOT}.") if(NOT EXISTS ${XCODE_ISYSROOT}) set(XCODE_ISYSROOT ${APPLE_DEV_DIR}/Platforms/MacOSX.platform/Developer/SDKs/MacOSX.sdk) if(NOT EXISTS ${XCODE_ISYSROOT}) message(FATAL_ERROR "Cannot find macos sdk.") endif() endif() endif() message(STATUS "Setting CMAKE_OSX_SYSROOT for macos ${OSX_SHORT_VERSION} to ${XCODE_ISYSROOT}") set(CMAKE_OSX_SYSROOT ${XCODE_ISYSROOT}) endif() endif() if (NGRAPH_MLIR_ENABLE) #MLIR uses the latest version of LLVM which requires C++14. set(NGRAPH_CXX_STANDARD 14) else() set(NGRAPH_CXX_STANDARD 11) endif() set(CMAKE_CXX_STANDARD ${NGRAPH_CXX_STANDARD}) set(CMAKE_CXX_STANDARD_REQUIRED ON) set(CMAKE_CXX_EXTENSIONS OFF) # Create compilation database compile_commands.json set(CMAKE_EXPORT_COMPILE_COMMANDS ON) set(CMAKE_POSITION_INDEPENDENT_CODE ON) # Setup CMAKE_ARGS to be forwarded to External Projects set(NGRAPH_FORWARD_CMAKE_ARGS -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER} -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER} -DCMAKE_CXX_STANDARD:STRING=${CMAKE_CXX_STANDARD} -DCMAKE_CXX_STANDARD_REQUIRED:BOOL=${CMAKE_CXX_STANDARD_REQUIRED} -DCMAKE_CXX_EXTENSIONS:BOOL=${CMAKE_CXX_EXTENSIONS} -DCMAKE_EXPORT_COMPILE_COMMANDS:BOOL=${CMAKE_EXPORT_COMPILE_COMMANDS} -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=${CMAKE_POSITION_INDEPENDENT_CODE} ) if (CMAKE_OSX_SYSROOT) set(NGRAPH_FORWARD_CMAKE_ARGS ${NGRAPH_FORWARD_CMAKE_ARGS} -DCMAKE_OSX_SYSROOT=${CMAKE_OSX_SYSROOT} ) endif() if (NOT MSVC) if(NOT CMAKE_BUILD_TYPE) set(CMAKE_BUILD_TYPE "Release" CACHE STRING "Build type" FORCE) endif() set(NGRAPH_FORWARD_CMAKE_ARGS ${NGRAPH_FORWARD_CMAKE_ARGS} -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE} ) endif() message(STATUS "NGRAPH_FORWARD_CMAKE_ARGS ${NGRAPH_FORWARD_CMAKE_ARGS}") # Prevent Eigen from using any LGPL3 code set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DEIGEN_MPL2_ONLY -DTBB_USE_THREADING_TOOLS") ngraph_var(NGRAPH_USE_PREBUILT_MLIR DEFAULT "FALSE") ngraph_var(NGRAPH_USE_PREBUILT_LLVM DEFAULT "FALSE") option(NGRAPH_UNIT_TEST_ENABLE "Control the building of unit tests" TRUE) option(NGRAPH_DOC_BUILD_ENABLE "Control the building of documentation" FALSE) option(NGRAPH_TOOLS_ENABLE "Control the building of tool" TRUE) option(NGRAPH_CPU_ENABLE "Control the building of the CPU backend" TRUE) option(NGRAPH_USE_LEGACY_MKLDNN "Use legacy MKLDNN" FALSE) option(NGRAPH_MLIR_ENABLE "Control the building of MLIR backend" FALSE) option(NGRAPH_INTERPRETER_ENABLE "Control the building of the INTERPRETER backend" TRUE) option(NGRAPH_NOP_ENABLE "Control the building of the NOP backend" TRUE) option(NGRAPH_GENERIC_CPU_ENABLE "Enable build nGraph for generic CPU backend" FALSE) option(NGRAPH_DEBUG_ENABLE "Enable output for NGRAPH_DEBUG statements" FALSE) option(NGRAPH_DEPRECATED_ENABLE "Enable compiler deprecation pragmas for deprecated APIs (recommended only for development use)" FALSE) option(NGRAPH_ONNX_IMPORT_ENABLE "Enable ONNX importer" FALSE) option(NGRAPH_DEX_ONLY "Build CPU DEX without codegen" FALSE) option(NGRAPH_ENABLE_CPU_CONV_AUTO "Enable mkldnn convolution_auto for CPU" TRUE) option(NGRAPH_CODE_COVERAGE_ENABLE "Enable code coverage data collection" FALSE) option(NGRAPH_LIB_VERSIONING_ENABLE "Enable shared library versioning" FALSE) option(NGRAPH_PYTHON_BUILD_ENABLE "Enable build nGraph python package wheel" FALSE) option(NGRAPH_PLAIDML_ENABLE "Enable the PlaidML backend" ${PLAIDML_FOUND}) option(NGRAPH_DISTRIBUTED_ENABLE "Enable distributed training using MLSL/OpenMPI" OFF) option(NGRAPH_FAST_MATH_ENABLE "Enable fast math" ON) option(NGRAPH_JSON_ENABLE "Enable JSON based serialization and tracing features" TRUE) option(NGRAPH_STATIC_LIB_ENABLE "Enable build nGraph as a static library" FALSE) option(NGRAPH_INTERPRETER_STATIC_LIB_ENABLE "Enable build INTERPRETER backend as a static library" FALSE) option(NGRAPH_CPU_STATIC_LIB_ENABLE "Enable build CPU backend as a static library" FALSE) option(NGRAPH_PLAIDML_STATIC_LIB_ENABLE "Enable build PlaidML backend as a static library" FALSE) option(NGRAPH_DYNAMIC_COMPONENTS_ENABLE "Enable dynamic loading of components" TRUE) option(NGRAPH_NATIVE_ARCH_ENABLE "Enable build for native archtecture" TRUE) if (NGRAPH_CPU_ENABLE) option(NGRAPH_TBB_ENABLE "Control usage of TBB for CPU backend" TRUE) endif() if (NGRAPH_CPU_ENABLE AND (NOT NGRAPH_GENERIC_CPU_ENABLE) ) set(NGRAPH_INTEL_CPU_ONLY_ENABLE ON) endif() if (NGRAPH_MLIR_ENABLE AND (NOT NGRAPH_DEX_ONLY)) #disable code-gen due to sym collision with LLVM message(FATAL_ERROR "Cannot build with both MLIR and code-gen ON.\n" "Use -DNGRAPH_DEX_ONLY=ON and try again. \n") endif() if (NGRAPH_DISTRIBUTED_ENABLE) if ("${NGRAPH_DISTRIBUTED_ENABLE}" STREQUAL "MLSL") if (NGRAPH_INTEL_CPU_ONLY_ENABLE) set(NGRAPH_DISTRIBUTED_MLSL_ENABLE TRUE) else() message(FATAL_ERROR "-DNGRAPH_DISTRIBUTED_ENABLE=MLSL to be used, if Intel CPU is the only backend enabled.\n" "Use -DNGRAPH_DISTRIBUTED_ENABLE=OMPI for all other situations.\n") endif() elseif("${NGRAPH_DISTRIBUTED_ENABLE}" STREQUAL "OMPI") set(NGRAPH_DISTRIBUTED_OMPI_ENABLE TRUE) else() message(FATAL_ERROR "Invalid arguments passed to NGRAPH_DISTRIBUTED_ENABLE, must select one of MLSL, OMPI or OFF.\n" "If using Intel CPU only backend, recommend Intel MLSL by setting -DNGRAPH_DISTRIBUTED_ENABLE=MLSL .\n") endif() endif() if (NGRAPH_ONNX_IMPORT_ENABLE) option(NGRAPH_USE_SYSTEM_PROTOBUF "Use system provided Protobuf shared object" FALSE) option(NGRAPH_ONNXIFI_ENABLE "Enable ONNX Interface for Framework Integration" TRUE) endif() if (NOT NGRAPH_JSON_ENABLE) set(NGRAPH_TOOLS_ENABLE FALSE) endif() macro (NORMALIZE_BOOL VAL) if (${VAL}) set(${VAL} ON) else() set(${VAL} OFF) endif() endmacro() NORMALIZE_BOOL(NGRAPH_UNIT_TEST_ENABLE) NORMALIZE_BOOL(NGRAPH_DOC_BUILD_ENABLE) NORMALIZE_BOOL(NGRAPH_TOOLS_ENABLE) NORMALIZE_BOOL(NGRAPH_CPU_ENABLE) NORMALIZE_BOOL(NGRAPH_USE_LEGACY_MKLDNN) NORMALIZE_BOOL(NGRAPH_MLIR_ENABLE) NORMALIZE_BOOL(NGRAPH_INTERPRETER_ENABLE) NORMALIZE_BOOL(NGRAPH_NOP_ENABLE) NORMALIZE_BOOL(NGRAPH_GENERIC_CPU_ENABLE) NORMALIZE_BOOL(NGRAPH_DEBUG_ENABLE) NORMALIZE_BOOL(NGRAPH_DEPRECATED_ENABLE) NORMALIZE_BOOL(NGRAPH_ONNX_IMPORT_ENABLE) NORMALIZE_BOOL(NGRAPH_DEX_ONLY) NORMALIZE_BOOL(NGRAPH_ENABLE_CPU_CONV_AUTO) NORMALIZE_BOOL(NGRAPH_CODE_COVERAGE_ENABLE) NORMALIZE_BOOL(NGRAPH_LIB_VERSIONING_ENABLE) NORMALIZE_BOOL(NGRAPH_PYTHON_BUILD_ENABLE) NORMALIZE_BOOL(NGRAPH_USE_PREBUILT_LLVM) NORMALIZE_BOOL(NGRAPH_USE_PREBUILT_MLIR) NORMALIZE_BOOL(NGRAPH_PLAIDML_ENABLE) NORMALIZE_BOOL(NGRAPH_JSON_ENABLE) NORMALIZE_BOOL(NGRAPH_NATIVE_ARCH_ENABLE) NORMALIZE_BOOL(NGRAPH_STATIC_LIB_ENABLE) NORMALIZE_BOOL(NGRAPH_INTERPRETER_STATIC_LIB_ENABLE) NORMALIZE_BOOL(NGRAPH_CPU_STATIC_LIB_ENABLE) NORMALIZE_BOOL(NGRAPH_DYNAMIC_COMPONENTS_ENABLE) if (NGRAPH_CPU_ENABLE) NORMALIZE_BOOL(NGRAPH_TBB_ENABLE) endif() message(STATUS "NGRAPH_CXX_STANDARD: ${NGRAPH_CXX_STANDARD}") message(STATUS "NGRAPH_UNIT_TEST_ENABLE: ${NGRAPH_UNIT_TEST_ENABLE}") message(STATUS "NGRAPH_DOC_BUILD_ENABLE: ${NGRAPH_DOC_BUILD_ENABLE}") message(STATUS "NGRAPH_TOOLS_ENABLE: ${NGRAPH_TOOLS_ENABLE}") message(STATUS "NGRAPH_CPU_ENABLE: ${NGRAPH_CPU_ENABLE}") message(STATUS "NGRAPH_USE_LEGACY_MKLDNN: ${NGRAPH_USE_LEGACY_MKLDNN}") message(STATUS "NGRAPH_MLIR_ENABLE: ${NGRAPH_MLIR_ENABLE}") message(STATUS "NGRAPH_INTERPRETER_ENABLE: ${NGRAPH_INTERPRETER_ENABLE}") message(STATUS "NGRAPH_NOP_ENABLE: ${NGRAPH_NOP_ENABLE}") message(STATUS "NGRAPH_GENERIC_CPU_ENABLE: ${NGRAPH_GENERIC_CPU_ENABLE}") message(STATUS "NGRAPH_DEBUG_ENABLE: ${NGRAPH_DEBUG_ENABLE}") message(STATUS "NGRAPH_DEPRECATED_ENABLE: ${NGRAPH_DEPRECATED_ENABLE}") message(STATUS "NGRAPH_ONNX_IMPORT_ENABLE: ${NGRAPH_ONNX_IMPORT_ENABLE}") message(STATUS "NGRAPH_DEX_ONLY: ${NGRAPH_DEX_ONLY}") message(STATUS "NGRAPH_ENABLE_CPU_CONV_AUTO: ${NGRAPH_ENABLE_CPU_CONV_AUTO}") message(STATUS "NGRAPH_CODE_COVERAGE_ENABLE: ${NGRAPH_CODE_COVERAGE_ENABLE}") message(STATUS "NGRAPH_LIB_VERSIONING_ENABLE: ${NGRAPH_LIB_VERSIONING_ENABLE}") message(STATUS "NGRAPH_PYTHON_BUILD_ENABLE: ${NGRAPH_PYTHON_BUILD_ENABLE}") message(STATUS "NGRAPH_USE_PREBUILT_LLVM: ${NGRAPH_USE_PREBUILT_LLVM}") message(STATUS "NGRAPH_USE_PREBUILT_MLIR: ${NGRAPH_USE_PREBUILT_MLIR}") message(STATUS "NGRAPH_PLAIDML_ENABLE: ${NGRAPH_PLAIDML_ENABLE}") message(STATUS "NGRAPH_DISTRIBUTED_ENABLE: ${NGRAPH_DISTRIBUTED_ENABLE}") message(STATUS "NGRAPH_JSON_ENABLE: ${NGRAPH_JSON_ENABLE}") message(STATUS "NGRAPH_STATIC_LIB_ENABLE: ${NGRAPH_STATIC_LIB_ENABLE}") message(STATUS "NGRAPH_INTERPRETER_STATIC_LIB_ENABLE: ${NGRAPH_INTERPRETER_STATIC_LIB_ENABLE}") message(STATUS "NGRAPH_CPU_STATIC_LIB_ENABLE: ${NGRAPH_CPU_STATIC_LIB_ENABLE}") message(STATUS "NGRAPH_PLAIDML_STATIC_LIB_ENABLE: ${NGRAPH_PLAIDML_STATIC_LIB_ENABLE}") message(STATUS "NGRAPH_DYNAMIC_COMPONENTS_ENABLE: ${NGRAPH_DYNAMIC_COMPONENTS_ENABLE}") message(STATUS "NGRAPH_NATIVE_ARCH_ENABLE: ${NGRAPH_NATIVE_ARCH_ENABLE}") if (NGRAPH_CPU_ENABLE) message(STATUS "NGRAPH_TBB_ENABLE: ${NGRAPH_TBB_ENABLE}") endif() if("${CMAKE_GENERATOR}" STREQUAL "Ninja" AND (NGRAPH_TOOLS_ENABLE OR NGRAPH_CPU_ENABLE OR NGRAPH_MLIR_ENABLE OR NGRAPH_INTELGPU_ENABLE OR NGRAPH_GPU_ENABLE OR NGRAPH_INTERPRETER_ENABLE OR NGRAPH_NOP_ENABLE OR NGRAPH_GPUH_ENABLE OR NGRAPH_GENERIC_CPU_ENABLE OR NGRAPH_ONNX_IMPORT_ENABLE OR NGRAPH_PLAIDML_ENABLE OR NGRAPH_JSON_ENABLE)) message(FATAL_ERROR "Ninja generator not supported.") endif() #----------------------------------------------------------------------------------------------- # Installation logic... #----------------------------------------------------------------------------------------------- if (LINUX) include(GNUInstallDirs) else() set(CMAKE_INSTALL_BINDIR "bin") set(CMAKE_INSTALL_INCLUDEDIR "include") set(CMAKE_INSTALL_DOCDIR "doc") set(CMAKE_INSTALL_LIBDIR "lib") endif() if (DEFINED NGRAPH_INSTALL_PREFIX) set(CMAKE_INSTALL_PREFIX ${NGRAPH_INSTALL_PREFIX}) endif() message(STATUS "Installation directory: ${CMAKE_INSTALL_PREFIX}") # Destinations set(NGRAPH_INSTALL_LIB "${CMAKE_INSTALL_LIBDIR}") set(NGRAPH_INSTALL_INCLUDE "${CMAKE_INSTALL_INCLUDEDIR}") set(NGRAPH_INSTALL_DOC "${CMAKE_INSTALL_DOCDIR}") set(NGRAPH_INSTALL_BIN "${CMAKE_INSTALL_BINDIR}") if (LINUX) if (DEFINED NGRAPH_RPATH) set(CMAKE_BUILD_RPATH "$ORIGIN:${NGRAPH_RPATH}") set(CMAKE_INSTALL_RPATH "$ORIGIN:${NGRAPH_RPATH}") else() set(CMAKE_BUILD_RPATH "$ORIGIN") set(CMAKE_INSTALL_RPATH "$ORIGIN") endif() endif() #----------------------------------------------------------------------------------------------- # Compiler-specific logic... #----------------------------------------------------------------------------------------------- # Compiler-specific logic... if ("${CMAKE_CXX_COMPILER_ID}" MATCHES "^(Apple)?Clang$" AND NOT NGRAPH_BUILD_DIR) message( STATUS "Setting clang flags...") include( cmake/clang_4_0_flags.cmake ) endif() if (WIN32) string(REPLACE "/W3" "/W0" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /EHsc /MP") if (CMAKE_CXX_COMPILER_ID MATCHES MSVC) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /bigobj") # C4251 needs to have dll-interface to be used by clients of class set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /wd4251") # C4275 non dll-interface class used as base for dll-interface class set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /wd4275") endif() else() set(CMAKE_CXX_FLAGS_RELWITHDEBINFO "-g") set(CMAKE_CXX_FLAGS_DEBUG "-O0 -g") # These can be uncommented once we have visibility fully in place # set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden") # set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility-inlines-hidden") endif() include(cmake/sdl.cmake) if (CMAKE_CXX_COMPILER_ID STREQUAL "GNU") if (DEFINED NGRAPH_USE_CXX_ABI) message( STATUS "nGraph using CXX11 ABI: " ${NGRAPH_USE_CXX_ABI} ) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D_GLIBCXX_USE_CXX11_ABI=${NGRAPH_USE_CXX_ABI}") endif() endif() set(CMAKE_ORIGINAL_CXX_FLAGS "${CMAKE_CXX_FLAGS}") ngraph_var(NGRAPH_WARNINGS_AS_ERRORS DEFAULT "OFF") if (${NGRAPH_WARNINGS_AS_ERRORS}) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Werror") message(STATUS "Warnings as errors") endif() if (NGRAPH_CODE_COVERAGE_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} --coverage") endif() # Enable build target CPU features if(NOT WIN32 AND NGRAPH_NATIVE_ARCH_ENABLE) set(NGRAPH_TARGET_ARCH native CACHE STRING "Target CPU architecture to build for. Defaults to the native CPU architecture") if (NOT "${NGRAPH_TARGET_ARCH}" STREQUAL "native") message(WARNING "Build target architecture was overridden. The resulting build might not work correctly on the host CPU.") endif() set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=${NGRAPH_TARGET_ARCH}") endif() if (DEFINED NGRAPH_TUNE_ARCH) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -mtune=${NGRAPH_TUNE_ARCH}") endif() if (NGRAPH_USE_GOLD) set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -fuse-ld=gold") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fuse-ld=gold") endif() if(WIN32) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNOMINMAX") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D_CRT_SECURE_NO_WARNINGS") set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS TRUE) endif() if (NGRAPH_CPU_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_CPU_ENABLE") endif() if (NGRAPH_USE_LEGACY_MKLDNN) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_USE_LEGACY_MKLDNN") endif() if (NGRAPH_MLIR_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_MLIR_ENABLE") set(NGRAPH_MLIR_SOURCE_DIR ${CMAKE_SOURCE_DIR}/src/contrib/mlir) endif() if (NGRAPH_STATIC_LIB_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_STATIC_LIB_ENABLE") endif() if (NGRAPH_DYNAMIC_COMPONENTS_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_DYNAMIC_COMPONENTS_ENABLE") endif() if (NGRAPH_PLAIDML_ENABLE) find_package(PlaidML CONFIG) if (NOT PLAIDML_FOUND) message(FATAL_ERROR "PlaidML not installed") endif() set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_PlaidML_ENABLE") endif() if (NGRAPH_DISTRIBUTED_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_DISTRIBUTED_ENABLE") if (NGRAPH_DISTRIBUTED_MLSL_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_DISTRIBUTED_MLSL_ENABLE") elseif (NGRAPH_DISTRIBUTED_OMPI_ENABLE) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DNGRAPH_DISTRIBUTED_OMPI_ENABLE") endif() endif() if (NOT DEFINED NGRAPH_TBB_ENABLE) set(NGRAPH_TBB_ENABLE ${NGRAPH_CPU_ENABLE}) endif() # Since UNIX support Bash we can use a Bash script to do the clang-format functions # This is much faster than the cmake method if (UNIX) add_custom_target(style-check COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/maint/check-code-format.sh) add_custom_target(style-apply COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/maint/apply-code-format.sh) add_custom_target(style COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/maint/apply-code-format.sh) else() add_custom_target(style-check COMMAND ${CMAKE_COMMAND} -DNGRAPH_SOURCE_DIR="${CMAKE_CURRENT_SOURCE_DIR}" -P ${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules/style_check.cmake ) add_custom_target(style-apply COMMAND ${CMAKE_COMMAND} -DNGRAPH_SOURCE_DIR="${CMAKE_CURRENT_SOURCE_DIR}" -P ${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules/style_apply.cmake ) add_custom_target(style COMMAND ${CMAKE_COMMAND} -DNGRAPH_SOURCE_DIR="${CMAKE_CURRENT_SOURCE_DIR}" -P ${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules/style_apply.cmake ) endif() add_custom_target(fix-mode COMMAND ${CMAKE_COMMAND} -DNGRAPH_SOURCE_DIR="${CMAKE_CURRENT_SOURCE_DIR}" -P ${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules/fix_mode.cmake ) if (NGRAPH_USE_PREBUILT_MLIR) if (NOT DEFINED MLIR_LLVM_PREBUILT_PATH) message(FATAL_ERROR "MLIR_LLVM_PREBUILT_PATH not specified") endif() endif() #----------------------------------------------------------------------------------------------- # enable or disable output from NGRAPH_DEBUG statements #----------------------------------------------------------------------------------------------- if(NGRAPH_DEBUG_ENABLE) add_definitions(-DNGRAPH_DEBUG_ENABLE) endif() #----------------------------------------------------------------------------------------------- # enable or disable deprecation warnings for old APIs #----------------------------------------------------------------------------------------------- if(NGRAPH_DEPRECATED_ENABLE) add_definitions(-DNGRAPH_DEPRECATED_ENABLE) endif() add_definitions(-DPROJECT_ROOT_DIR="${CMAKE_CURRENT_SOURCE_DIR}") #----------------------------------------------------------------------------------------------- # Print Global Options #----------------------------------------------------------------------------------------------- message(STATUS "Compile Flags: ${CMAKE_CXX_FLAGS}") message(STATUS "Shared Link Flags: ${CMAKE_SHARED_LINKER_FLAGS}") message(STATUS "CMAKE_CXX_FLAGS_RELEASE ${CMAKE_CXX_FLAGS_RELEASE}") message(STATUS "CMAKE_CXX_FLAGS_DEBUG ${CMAKE_CXX_FLAGS_DEBUG}") #----------------------------------------------------------------------------------------------- # External projects install directory #----------------------------------------------------------------------------------------------- if (NOT NGRAPH_BUILD_DIR) set(NGRAPH_BUILD_DIR ${CMAKE_CURRENT_BINARY_DIR}/src/ngraph) set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${NGRAPH_BUILD_DIR}) if(WIN32) set(NGRAPH_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/${CMAKE_CFG_INTDIR}) set(NGRAPH_LIBRARY_INSTALL_SRC_DIRECTORY ${NGRAPH_BUILD_DIR}/\${CMAKE_INSTALL_CONFIG_NAME}) set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${NGRAPH_BUILD_DIR}) set(NGRAPH_ARCHIVE_OUTPUT_DIRECTORY ${NGRAPH_BUILD_DIR}/${CMAKE_CFG_INTDIR}) set(NGRAPH_ARCHIVE_INSTALL_SRC_DIRECTORY ${NGRAPH_BUILD_DIR}/\${CMAKE_INSTALL_CONFIG_NAME}) set(CMAKE_PDB_OUTPUT_DIRECTORY ${NGRAPH_BUILD_DIR}) set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${NGRAPH_BUILD_DIR}) else() set(NGRAPH_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}) set(NGRAPH_LIBRARY_INSTALL_SRC_DIRECTORY ${NGRAPH_BUILD_DIR}) endif() endif() # Build destination directory for nGraph binaries and tools. set(NGRAPH_BUILD_BIN ${CMAKE_CURRENT_BINARY_DIR}/bin) set(EXTERNAL_INSTALL_DIR ${CMAKE_BINARY_DIR}/external) if(NOT DEFINED EXTERNAL_PROJECTS_ROOT) set(EXTERNAL_PROJECTS_ROOT ${CMAKE_CURRENT_BINARY_DIR}) endif() if (NGRAPH_ONNX_IMPORT_ENABLE) if (MSVC) # When we build dll libraries. These flags make sure onnx and protobuf build with /MD, not /MT. # These two options can't be mixed, because they requires link two imcompatiable runtime. set(ONNX_USE_MSVC_STATIC_RUNTIME OFF) set(protobuf_WITH_ZLIB OFF CACHE BOOL "" FORCE) set(protobuf_MSVC_STATIC_RUNTIME OFF CACHE BOOL "Link protobuf to static runtime libraries" FORCE) endif() if (NOT NGRAPH_USE_SYSTEM_PROTOBUF) include(cmake/external_protobuf.cmake) else() find_package(Protobuf 2.6.1 REQUIRED) endif() include(cmake/external_onnx.cmake) if (TARGET ext_protobuf) add_dependencies(ext_onnx ext_protobuf) endif() endif() include(cmake/external_gtest.cmake) if(NGRAPH_JSON_ENABLE) include(cmake/external_json.cmake) endif() if(NGRAPH_CPU_ENABLE OR NGRAPH_GENERIC_CPU_ENABLE) include(cmake/external_eigen.cmake) endif() if(NGRAPH_CPU_ENABLE) if(NGRAPH_USE_LEGACY_MKLDNN) include(cmake/external_mkldnn.cmake) set(NGRAPH_DEX_ONLY TRUE) else() include(cmake/external_mkldnn_v1.cmake) endif() endif() if (NGRAPH_MLIR_ENABLE) include(cmake/external_mlir.cmake) endif() if (NGRAPH_CPU_ENABLE AND NOT NGRAPH_DEX_ONLY) set(NGRAPH_CODEGEN_ENABLE TRUE) else() set(NGRAPH_CODEGEN_ENABLE FALSE) endif() if(NGRAPH_CODEGEN_ENABLE) if (NGRAPH_USE_PREBUILT_LLVM OR DEFINED LLVM_TARBALL_URL) include(cmake/external_llvm_prebuilt.cmake) elseif (NOT MSVC) include(cmake/external_llvm.cmake) else() message(FATAL_ERROR "CODEGEN is not supported on Windows!") endif() endif() if(NGRAPH_TBB_ENABLE) if (WIN32 OR APPLE) include(cmake/external_tbb_prebuilt.cmake) else() include(cmake/external_tbb.cmake) endif() endif() if (NGRAPH_DISTRIBUTED_MLSL_ENABLE) include(cmake/external_mlsl.cmake) endif() if (NGRAPH_HALIDE) message(WARNING "Halide build system integration is currently using an older LLVM release \ and is not expected to work across most build environments. Consider \ disabling it till this message goes away") include(cmake/external_halide.cmake) endif() add_subdirectory(src) add_subdirectory(test) add_subdirectory(doc/examples) if (NGRAPH_DOC_BUILD_ENABLE) add_subdirectory(doc) endif() if (NGRAPH_PYTHON_BUILD_ENABLE) add_subdirectory(python) endif() include(CMakePackageConfigHelpers) export(TARGETS ngraph NAMESPACE ngraph:: FILE "${CMAKE_CURRENT_BINARY_DIR}/ngraphTargets.cmake") install(EXPORT ngraphTargets FILE ngraphTargets.cmake NAMESPACE ngraph:: DESTINATION cmake COMPONENT ngraph) configure_package_config_file(${CMAKE_CURRENT_SOURCE_DIR}/cmake/share/ngraphConfig.cmake.in ${CMAKE_CURRENT_BINARY_DIR}/ngraphConfig.cmake INSTALL_DESTINATION cmake) write_basic_package_version_file(${CMAKE_CURRENT_BINARY_DIR}/ngraphConfigVersion.cmake VERSION ${NGRAPH_VERSION} COMPATIBILITY SameMajorVersion) install(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/licenses DESTINATION "." COMPONENT ngraph ) install(FILES ${CMAKE_CURRENT_SOURCE_DIR}/LICENSE DESTINATION "." COMPONENT ngraph) install(FILES ${CMAKE_CURRENT_BINARY_DIR}/VERSION DESTINATION "." COMPONENT ngraph) install(FILES ${CMAKE_CURRENT_BINARY_DIR}/ngraphConfig.cmake ${CMAKE_CURRENT_BINARY_DIR}/ngraphConfigVersion.cmake DESTINATION cmake COMPONENT ngraph)