|
|
| if(NOT CMAKE_VERSION VERSION_LESS "3.15") |
| # enable CMAKE_MSVC_RUNTIME_LIBRARY |
| cmake_policy(SET CMP0091 NEW) |
| endif() |
|
|
| project(pnnx) |
| cmake_minimum_required(VERSION 3.12) |
|
|
| if(POLICY CMP0074) |
| cmake_policy(SET CMP0074 NEW) |
| endif() |
|
|
| if(MSVC AND NOT CMAKE_VERSION VERSION_LESS "3.15") |
| option(PNNX_BUILD_WITH_STATIC_CRT "Enables use of statically linked CRT for statically linked pnnx" OFF) |
| if(PNNX_BUILD_WITH_STATIC_CRT) |
| # cmake before version 3.15 not work |
| set(CMAKE_MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>") |
| endif() |
| endif() |
|
|
| list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake") |
| include(PNNXPyTorch) |
|
|
| # c++14 is required for using torch headers |
| set(CMAKE_CXX_STANDARD 14) |
|
|
| #set(CMAKE_BUILD_TYPE debug) |
| #set(CMAKE_BUILD_TYPE relwithdebinfo) |
| #set(CMAKE_BUILD_TYPE release) |
|
|
| option(PNNX_COVERAGE "build for coverage" OFF) |
|
|
| #set(Torch_INSTALL_DIR "/home/nihui/.local/lib/python3.9/site-packages/torch" CACHE STRING "") |
| #set(Torch_INSTALL_DIR "/home/nihui/osd/pnnx/pytorch-v1.10.0/build/install" CACHE STRING "") |
| #set(Torch_INSTALL_DIR "/home/nihui/osd/pnnx/libtorch" CACHE STRING "") |
| set(TorchVision_INSTALL_DIR "/home/nihui/osd/vision/build/install" CACHE STRING "") |
|
|
| #set(Torch_DIR "${Torch_INSTALL_DIR}/share/cmake/Torch") |
| set(TorchVision_DIR "${TorchVision_INSTALL_DIR}/share/cmake/TorchVision") |
|
|
| find_package(Python3 COMPONENTS Interpreter Development) |
|
|
| PNNXProbeForPyTorchInstall() |
| find_package(Torch REQUIRED) |
|
|
| find_package(TorchVision QUIET) |
|
|
| message(STATUS "Torch_VERSION = ${Torch_VERSION}") |
| message(STATUS "Torch_VERSION_MAJOR = ${Torch_VERSION_MAJOR}") |
| message(STATUS "Torch_VERSION_MINOR = ${Torch_VERSION_MINOR}") |
| message(STATUS "Torch_VERSION_PATCH = ${Torch_VERSION_PATCH}") |
|
|
| if(Torch_VERSION VERSION_LESS "1.8") |
| message(FATAL_ERROR "pnnx only supports PyTorch >= 1.8") |
| endif() |
|
|
| if(TorchVision_FOUND) |
| message(STATUS "Building with TorchVision") |
| add_definitions(-DPNNX_TORCHVISION) |
| else() |
| message(WARNING "Building without TorchVision") |
| endif() |
|
|
| include_directories(${TORCH_INCLUDE_DIRS}) |
|
|
| add_subdirectory(src) |
|
|
| enable_testing() |
| add_subdirectory(tests) |
|
|
