95 lines
3.2 KiB
CMake
95 lines
3.2 KiB
CMake
# ******************************************************************************
|
|
# Copyright 2017-2021 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.13)
|
|
|
|
project (pyngraph)
|
|
|
|
if(NOT DEFINED OpenVINO_MAIN_SOURCE_DIR)
|
|
find_package(InferenceEngineDeveloperPackage QUIET)
|
|
find_package(ngraph REQUIRED)
|
|
endif()
|
|
|
|
if(ngraph_FOUND)
|
|
message("ngraph version = {${ngraph_VERSION}}")
|
|
endif()
|
|
|
|
include(FetchContent)
|
|
|
|
FetchContent_Declare(
|
|
pybind11
|
|
GIT_REPOSITORY "https://github.com/pybind/pybind11.git"
|
|
GIT_TAG "v2.5.0"
|
|
)
|
|
|
|
FetchContent_GetProperties(pybind11)
|
|
if(NOT pybind11_POPULATED)
|
|
FetchContent_Populate(pybind11)
|
|
add_subdirectory(${pybind11_SOURCE_DIR} ${pybind11_BINARY_DIR})
|
|
endif()
|
|
|
|
find_package(Python COMPONENTS Interpreter Development REQUIRED)
|
|
|
|
if(PYTHON_FOUND)
|
|
set(PYTHON_VERSION python${PYTHON_VERSION_MAJOR}.${PYTHON_VERSION_MINOR})
|
|
message("Python version={${Python_VERSION}}")
|
|
else()
|
|
message(FATAL_ERROR "Python was not found!")
|
|
endif()
|
|
|
|
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_ARCHIVE_OUTPUT_DIRECTORY}/${PYTHON_VERSION})
|
|
set(CMAKE_COMPILE_PDB_OUTPUT_DIRECTORY ${CMAKE_ARCHIVE_OUTPUT_DIRECTORY}/${PYTHON_VERSION})
|
|
set(CMAKE_PDB_OUTPUT_DIRECTORY ${CMAKE_ARCHIVE_OUTPUT_DIRECTORY}/${PYTHON_VERSION})
|
|
|
|
# compile options
|
|
|
|
if (CMAKE_CXX_COMPILER_ID STREQUAL "MSVC")
|
|
# disable warning: This operator was deprecated and will be removed with v0 operation.
|
|
add_compile_options(/wd4996)
|
|
elseif(CMAKE_CXX_COMPILER_ID STREQUAL "Clang")
|
|
add_compile_options(-Wno-deprecated-register)
|
|
elseif(CMAKE_CXX_COMPILER_ID STREQUAL "AppleClang")
|
|
add_link_options(-stdlib=libc++)
|
|
add_compile_options(-Wno-unused-value)
|
|
elseif(CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
|
|
# WA for GCC 7.5 "PYBIND11_NOINLINE inline" warning
|
|
add_compile_options(-Wno-error=attributes)
|
|
endif()
|
|
|
|
# create target
|
|
|
|
file(GLOB_RECURSE SOURCES src/pyngraph/*.cpp)
|
|
|
|
pybind11_add_module(_${PROJECT_NAME} MODULE ${SOURCES})
|
|
|
|
target_include_directories(_${PROJECT_NAME} PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/src")
|
|
target_link_libraries(_${PROJECT_NAME} PRIVATE ngraph::ngraph ngraph::onnx_importer)
|
|
|
|
if(OpenVINO_MAIN_SOURCE_DIR OR InferenceEngineDeveloperPackage_FOUND)
|
|
ie_cpack_add_component(pyngraph_${PYTHON_VERSION})
|
|
|
|
install(TARGETS _${PROJECT_NAME}
|
|
DESTINATION python/${PYTHON_VERSION}
|
|
COMPONENT pyngraph_${PYTHON_VERSION})
|
|
|
|
install(DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/src/ngraph
|
|
DESTINATION python/${PYTHON_VERSION}
|
|
COMPONENT pyngraph_${PYTHON_VERSION}
|
|
USE_SOURCE_PERMISSIONS)
|
|
|
|
ie_cpack(pyngraph_${PYTHON_VERSION})
|
|
endif()
|