module_diago/CMakeLists.txt at main · haozhihan/module_diago · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
cmake_minimum_required(VERSION 3.15...3.27)
project(
  ${SKBUILD_PROJECT_NAME}
  VERSION ${SKBUILD_PROJECT_VERSION}
  LANGUAGES CXX
)

set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC")

# 添加子目录
add_subdirectory(src)

set(SOURCES
    src/binding.cpp
)

# 查找 OpenBLAS 库
find_package(BLAS REQUIRED)

# 查找 LAPACK 库
find_package(LAPACK REQUIRED)

pybind11_add_module(_py_dav MODULE ${SOURCES})
target_include_directories(_py_dav PRIVATE src)
target_link_libraries(
    _py_dav
    PRIVATE
    pybind11::headers
    diago
    device
    ${OpenBLAS_LIBRARIES}
    ${LAPACK_LIBRARIES}
)

target_compile_definitions(_py_dav PRIVATE VERSION_INFO=${PROJECT_VERSION})

set(TARGET_PACK module_diago)
set(CMAKE_INSTALL_RPATH "${PYTHON_SITE_PACKAGES}/${TARGET_PACK}")
set_target_properties(_py_dav PROPERTIES INSTALL_RPATH "$ORIGIN")
set_target_properties(device PROPERTIES INSTALL_RPATH "$ORIGIN")
set_target_properties(diago PROPERTIES INSTALL_RPATH "$ORIGIN")

install(TARGETS _py_dav device DESTINATION ${TARGET_PACK})
install(TARGETS _py_dav diago DESTINATION ${TARGET_PACK})