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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
# Author: Lisandro Dalcin
# Contact: dalcinl@gmail.com
cmake_minimum_required(VERSION 3.15...3.26)
project(mpi4py LANGUAGES C)
set(
MPIABI $ENV{MPI4PY_BUILD_MPIABI} CACHE BOOL
"Use MPI ABI (standard or legacy)"
)
set(
PYSABI $ENV{MPI4PY_BUILD_PYSABI} CACHE STRING
"Use Python Stable ABI"
)
if (PYSABI)
if (NOT SKBUILD_SABI_COMPONENT)
set(SKBUILD_SABI_COMPONENT Development.SABIModule)
endif()
if (NOT SKBUILD_SABI_VERSION)
set(SKBUILD_SABI_VERSION ${PYSABI})
endif()
endif()
if (SKBUILD_SABI_VERSION)
set(WITH_SABI USE_SABI ${SKBUILD_SABI_VERSION})
endif()
find_package(
Python REQUIRED
COMPONENTS Interpreter
COMPONENTS Development.Module ${SKBUILD_SABI_COMPONENT}
OPTIONAL_COMPONENTS Development.Embed
)
find_package(MPI REQUIRED)
separate_arguments(MPI_C_COMPILE_OPTIONS NATIVE_COMMAND)
separate_arguments(MPI_C_LINK_FLAGS NATIVE_COMMAND)
include(CheckSymbolExists)
set(CMAKE_REQUIRED_INCLUDES ${MPI_C_INCLUDE_DIRS})
set(CMAKE_REQUIRED_DEFINITIONS ${MPI_C_COMPILE_DEFINITIONS})
set(CMAKE_REQUIRED_FLAGS ${MPI_C_COMPILE_OPTIONS})
set(BINDIR ${CMAKE_CURRENT_BINARY_DIR})
set(TOPDIR ${CMAKE_CURRENT_SOURCE_DIR})
set(SRCDIR ${CMAKE_CURRENT_SOURCE_DIR}/src)
# Cython
set(cythonize ${TOPDIR}/conf/cythonize.py)
set(Cython_COMMAND ${Python_EXECUTABLE} ${cythonize})
set(Cython_OPTIONS --3str --cleanup 3)
set(MPI.pyx ${SRCDIR}/mpi4py/MPI.pyx)
file(
GLOB_RECURSE MPI.deps
${SRCDIR}/mpi4py/*.pyx
${SRCDIR}/mpi4py/*.pxd
${SRCDIR}/mpi4py/MPI.src/*.pyx
${SRCDIR}/mpi4py/MPI.src/*.pxi
)
set(MPI.c ${BINDIR}/MPI.c)
set(MPI.h ${BINDIR}/MPI.h ${BINDIR}/MPI_api.h)
add_custom_command(
OUTPUT ${MPI.c}
BYPRODUCTS ${MPI.h}
DEPENDS ${MPI.deps}
WORKING_DIRECTORY ${TOPDIR}
VERBATIM
COMMAND
${Cython_COMMAND} ${Cython_OPTIONS}
${MPI.pyx} --output-file ${MPI.c}
)
install(FILES ${MPI.h} DESTINATION mpi4py)
# mpi4py.MPI
python_add_library(mpi4py.MPI MODULE ${MPI.c} WITH_SOABI ${WITH_SABI})
set_target_properties(mpi4py.MPI PROPERTIES OUTPUT_NAME "MPI" PREFIX "")
target_compile_definitions(mpi4py.MPI PRIVATE $<$<BOOL:${MPIABI}>:PYMPIABI=1>)
target_include_directories(mpi4py.MPI PRIVATE ${SRCDIR})
target_include_directories(mpi4py.MPI PRIVATE ${MPI_C_INCLUDE_DIRS})
target_compile_definitions(mpi4py.MPI PRIVATE ${MPI_C_COMPILE_DEFINITIONS})
target_compile_options(mpi4py.MPI PRIVATE ${MPI_C_COMPILE_OPTIONS})
target_link_directories(mpi4py.MPI PRIVATE ${MPI_C_LIBRARY_DIRS})
target_link_libraries(mpi4py.MPI PRIVATE ${MPI_C_LIBRARIES})
target_link_options(mpi4py.MPI PRIVATE ${MPI_C_LINK_FLAGS})
install(TARGETS mpi4py.MPI LIBRARY DESTINATION mpi4py)
# mpi4py/bin/python-mpi
if (Python_Development.Embed_FOUND)
add_executable(python-mpi ${SRCDIR}/python.c)
target_include_directories(python-mpi PRIVATE ${Python_INCLUDE_DIRS})
target_link_directories(python-mpi PRIVATE ${Python_LIBRARY_DIRS})
target_link_libraries(python-mpi PRIVATE ${Python_LIBRARIES})
target_include_directories(python-mpi PRIVATE ${MPI_C_INCLUDE_DIRS})
target_compile_definitions(python-mpi PRIVATE ${MPI_C_COMPILE_DEFINITIONS})
target_compile_options(python-mpi PRIVATE ${MPI_C_COMPILE_OPTIONS})
target_link_directories(python-mpi PRIVATE ${MPI_C_LIBRARY_DIRS})
target_link_libraries(python-mpi PRIVATE ${MPI_C_LIBRARIES})
target_link_options(python-mpi PRIVATE ${MPI_C_LINK_FLAGS})
install(TARGETS python-mpi RUNTIME DESTINATION mpi4py/bin)
endif()
# mpi4py/*
file(
GLOB mpi4py_SOURCES
RELATIVE ${SRCDIR}
${SRCDIR}/mpi4py/*.py
${SRCDIR}/mpi4py/futures/*.py
${SRCDIR}/mpi4py/util/*.py
)
file(
GLOB mpi4py_HEADERS
RELATIVE ${SRCDIR}
${SRCDIR}/mpi4py/*.pxd
${SRCDIR}/mpi4py/include/mpi4py/*.[hi]
${SRCDIR}/mpi4py/include/mpi4py/*.px[di]
)
if (TRUE)
list(APPEND mpi4py_HEADERS mpi4py/py.typed)
list(APPEND mpi4py_HEADERS mpi4py/MPI.pyi)
foreach(file ${mpi4py_SOURCES})
list(APPEND mpi4py_HEADERS ${file}i)
endforeach()
endif()
foreach(file ${mpi4py_SOURCES} ${mpi4py_HEADERS})
get_filename_component(dir ${file} DIRECTORY)
install(FILES ${SRCDIR}/${file} DESTINATION ${dir})
endforeach()
if (WIN32)
set(mpidllpath _mpi_dll_path.py mpi.pth)
foreach(file ${mpidllpath})
if (EXISTS ${SRCDIR}/${file})
install(FILES ${SRCDIR}/${file} DESTINATION .)
endif()
endforeach()
endif()
|
