# ****************************************************************************** # Copyright 2017-2018 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.1) set(NGRAPH_INCLUDE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/src ) # Suppress an OS X-specific warning. if (POLICY CMP0042) cmake_policy(SET CMP0042 OLD) endif() project (ngraph) SET(GCC_MIN_VERSION 4.8) SET(CLANG_MIN_VERSION 3.8) SET(APPLE_CLANG_MIN_VERSION 9.0) if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU") if (CMAKE_CXX_COMPILER_VERSION VERSION_LESS GCC_MIN_VERSION) message(FATAL_ERROR "GCC version must be at least ${GCC_MIN_VERSION}!") endif() elseif ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang") if (CMAKE_CXX_COMPILER_VERSION VERSION_LESS CLANG_MIN_VERSION) message(FATAL_ERROR "Clang version must be at least ${CLANG_MIN_VERSION}!") endif() elseif ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "AppleClang") if (CMAKE_CXX_COMPILER_VERSION VERSION_LESS APPLE_CLANG_MIN_VERSION) message(FATAL_ERROR "Apple Clang version must be at least ${APPLE_CLANG_MIN_VERSION}!") endif() else() message(WARNING "You are using an unsupported compiler. Compilation has only been tested with Clang (${CLANG_MIN_VERSION} and up), Apple Clang (${APPLE_CLANG_MIN_VERSION} and up), and GCC (${GCC_MIN_VERSION} and up).") endif() if($ENV{NGRAPH_USE_PREBUILT_LLVM}) set(NGRAPH_USE_PREBUILT_LLVM TRUE) endif() # These variables are undocumented but useful. set(CMAKE_DISABLE_SOURCE_CHANGES ON) set(CMAKE_DISABLE_IN_SOURCE_BUILD ON) # Create compilation database compile_commands.json set(CMAKE_EXPORT_COMPILE_COMMANDS ON) # set directory where the custom finders live set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/cmake/Modules/") include(var_functions) #----------------------------------------------------------------------------------------------- # Installation logic... #----------------------------------------------------------------------------------------------- # Default installation location for cmake usually is /usr/include or /usr/local/include # which requires sudo access on most servers. Also, this creates a problem for the shared # development systems (e.g., build servers). # # Therefore we are setting the installation directory so that by defult "make install" # won't override artifacts generated by other users. # # The user can always override this by using the cmake command listed below. if (DEFINED NGRAPH_INSTALL_PREFIX) set(CMAKE_INSTALL_PREFIX ${NGRAPH_INSTALL_PREFIX}) else() set(CMAKE_INSTALL_PREFIX "$ENV{HOME}/ngraph_dist") endif() message (STATUS "Installation directory: ${CMAKE_INSTALL_PREFIX}") message (STATUS "To Override use: cmake -DNGRAPH_INSTALL_PREFIX=/foo") # Destinations set(NGRAPH_INSTALL_LIB "${CMAKE_INSTALL_PREFIX}/lib") set(NGRAPH_INSTALL_INCLUDE "${CMAKE_INSTALL_PREFIX}/include") set(NGRAPH_INSTALL_DOC "${CMAKE_INSTALL_PREFIX}/doc") #----------------------------------------------------------------------------------------------- # Compiler-specific logic... #----------------------------------------------------------------------------------------------- # Compiler-specific logic... if ("${CMAKE_CXX_COMPILER_ID}" MATCHES "^(Apple)?Clang$") message( STATUS "Setting clang flags...") include( cmake/clang_4_0_flags.cmake ) endif() set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3") ngraph_var(NGRAPH_WARNINGS_AS_ERRORS DEFAULT "OFF") if ("${NGRAPH_WARNINGS_AS_ERRORS}" MATCHES "^(ON|TRUE)$") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Werror") message(STATUS "Warnings as errors") endif() SET(CMAKE_CXX_FLAGS_RELWITHDEBINFO "-O3 -g") SET(CMAKE_CXX_FLAGS_RELEASE "-O3") SET(CMAKE_CXX_FLAGS_DEBUG "-O0 -g") # Set true if CPU backend is built by default if (NOT DEFINED NGRAPH_CPU_ENABLE) set(NGRAPH_CPU_ENABLE TRUE) endif() if (NOT DEFINED NGRAPH_TBB_ENABLE) set(NGRAPH_TBB_ENABLE ${NGRAPH_CPU_ENABLE}) endif() #----------------------------------------------------------------------------------------------- # GPU support #----------------------------------------------------------------------------------------------- # Setup CUDA and cuDNN if NGRAPH_GPU_ENABLE=TRUE find_package(CUDA 8 QUIET) if(CUDA_FOUND AND NGRAPH_GPU_ENABLE) message(STATUS "GPU Backend Enabled") set(NGRAPH_GPU_ENABLE TRUE) find_package(CUDNN 5 QUIET REQUIRED) include_directories(SYSTEM ${CUDA_INCLUDE_DIRS} ${CUDNN_INCLUDE_DIR} ${LLVM_INCLUDE_DIR}) if (CMAKE_C_COMPILER_ID STREQUAL "GNU" AND NOT CMAKE_C_COMPILER_VERSION VERSION_LESS 6.0 AND CUDA_HOST_COMPILER STREQUAL CMAKE_C_COMPILER) message(FATAL_ERROR "CUDA 8.0 is not compatible with GCC version >= 6.\n" "Please select a correct compiler version\n" ) endif() elseif(NGRAPH_GPU_ENABLE) message(FATAL_ERROR "GPU was required but CUDA library was not found") endif() #----------------------------------------------------------------------------------------------- # distributed support #----------------------------------------------------------------------------------------------- if(NGRAPH_DISTRIBUTED_ENABLE) find_package(MPI REQUIRED) if(MPI_CXX_FOUND) add_definitions(-DNGRAPH_DISTRIBUTED) endif() endif() #----------------------------------------------------------------------------------------------- # External projects install directory #----------------------------------------------------------------------------------------------- # Root for external installs, when using `make`, e.g. # ${EXTERNAL_PROJECTS_ROOT} # ├── eigen # │ ├── include # │ │ └── eigen3 <-- ${EIGEN_INCLUDE_DIR} # │ │ ├── Eigen # │ ... # │ # └── mkldnn # ├── include <-- ${MKLDNN_INCLUDE_DIR} # │ ├── mkldnn.h # │ ... # ├── lib <-- ${MKLDNN_LIB_DIR} # │ ├── libiomp5.so # ... set(EXTERNAL_INSTALL_DIR ${CMAKE_BINARY_DIR}/external) # The following 'add_subdirectory' call: # - defines the 'libgtest' target. # - defines the 'GTEST_INCLUDE_DIR' cmake variable. add_subdirectory(third-party) # The following 'add_subdirectory' call: # - defines the 'ngraph' library target. # - defines the 'NGRAPH_INCLUDE_DIR' cmake variable. # - defines the 'NGRAPH_VERSION' cmake variable. # - install-related targets. add_subdirectory(src) add_subdirectory(test) add_subdirectory(doc)