40 lines
1.3 KiB
CMake
40 lines
1.3 KiB
CMake
# Copyright (C) 2018-2023 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
# [cmake:extension]
|
|
set(CMAKE_CXX_STANDARD 11)
|
|
|
|
set(TARGET_NAME "template_extension")
|
|
|
|
find_package(ngraph REQUIRED OPTIONAL_COMPONENTS onnx_importer)
|
|
find_package(InferenceEngine REQUIRED)
|
|
find_package(OpenCV QUIET COMPONENTS core)
|
|
|
|
set(SRC cpu_kernel.cpp extension.cpp op.cpp)
|
|
|
|
if (OpenCV_FOUND)
|
|
set(SRC ${SRC} fft_kernel.cpp fft_op.cpp)
|
|
endif()
|
|
|
|
add_library(${TARGET_NAME} MODULE ${SRC})
|
|
|
|
if (OpenCV_FOUND)
|
|
target_compile_definitions(${TARGET_NAME} PRIVATE OPENCV_IMPORT_ENABLED)
|
|
target_link_libraries(${TARGET_NAME} PRIVATE opencv_core)
|
|
endif()
|
|
|
|
target_compile_definitions(${TARGET_NAME} PRIVATE IMPLEMENT_INFERENCE_EXTENSION_API)
|
|
target_link_libraries(${TARGET_NAME} PRIVATE IE::inference_engine
|
|
${NGRAPH_LIBRARIES})
|
|
|
|
if (ngraph_onnx_importer_FOUND)
|
|
target_link_libraries(${TARGET_NAME} PRIVATE ${ONNX_IMPORTER_LIBRARIES})
|
|
target_compile_definitions(${TARGET_NAME} PRIVATE NGRAPH_ONNX_IMPORT_ENABLED)
|
|
endif()
|
|
# [cmake:extension]
|
|
|
|
# Enable code style check
|
|
file(GLOB_RECURSE template_extension_src "${CMAKE_CURRENT_SOURCE_DIR}/*.cpp" "${CMAKE_CURRENT_SOURCE_DIR}/*.hpp")
|
|
add_clang_format_target(${TARGET_NAME}_clang FOR_SOURCES ${template_extension_src})
|