23 lines
667 B
CMake
23 lines
667 B
CMake
cmake_minimum_required(VERSION 3.5)
|
|
project(PyWrap)
|
|
|
|
set(CMAKE_CXX_STANDARD 20)
|
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
|
|
|
find_package(Python3 3.11...3.11.9 COMPONENTS Interpreter Development REQUIRED)
|
|
find_package(Torch REQUIRED)
|
|
find_package(Boost REQUIRED COMPONENTS python3 numpy3)
|
|
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
|
|
|
|
# Temporary patch while find_package(Torch) is not fixed
|
|
file(
|
|
GLOB
|
|
LIBTORCH_PYTHON
|
|
"${TORCH_INSTALL_PREFIX}/lib/libtorch_python.so"
|
|
"${TORCH_INSTALL_PREFIX}/lib/libtorch_python.dylib"
|
|
)
|
|
message(STATUS "TORCH Libraries: ${LIBTORCH_PYTHON}")
|
|
|
|
add_subdirectory(lib/Files)
|
|
add_subdirectory(src) |