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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
include(CheckFunctionExists)
set(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} -DDEBUG")
if(CMAKE_COMPILER_IS_GNUCC)
add_definitions(-Wdeclaration-after-statement)
endif()
include_directories("${CMAKE_CURRENT_SOURCE_DIR}")
add_custom_command(
OUTPUT ${CMAKE_CURRENT_SOURCE_DIR}/gpuarray_types.c
${CMAKE_CURRENT_SOURCE_DIR}/gpuarray/types.h
COMMAND python gen_types.py
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/gen_types.py)
add_custom_command(
OUTPUT ${CMAKE_CURRENT_SOURCE_DIR}/cluda_cuda.h.c
COMMAND python head.py cluda_cuda.h
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/head.py
${CMAKE_CURRENT_SOURCE_DIR}/cluda_cuda.h
)
add_custom_command(
OUTPUT ${CMAKE_CURRENT_SOURCE_DIR}/cluda_opencl.h.c
COMMAND python head.py cluda_opencl.h
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/head.py
${CMAKE_CURRENT_SOURCE_DIR}/cluda_opencl.h
)
macro (set_rel var)
file (RELATIVE_PATH _relPath "${CMAKE_SOURCE_DIR}/src" "${CMAKE_CURRENT_SOURCE_DIR}")
# clear previous list (if any)
set(SET_REL_TMP)
foreach (_src ${ARGN})
if (_relPath)
list(APPEND SET_REL_TMP "${_relPath}/${_src}")
else()
list(APPEND SET_REL_TMP "${_src}")
endif()
endforeach()
if (_relPath)
set(${var} ${SET_REL_TMP} PARENT_SCOPE)
else()
set(${var} ${SET_REL_TMP})
endif()
endmacro()
set(_GPUARRAY_SRC
cache/lru.c
cache/twoq.c
cache/disk.c
gpuarray_types.c
gpuarray_error.c
gpuarray_util.c
gpuarray_buffer.c
gpuarray_buffer_blas.c
gpuarray_buffer_collectives.c
gpuarray_array.c
gpuarray_array_blas.c
gpuarray_array_collectives.c
gpuarray_kernel.c
gpuarray_extension.c
gpuarray_elemwise.c
gpuarray_reduction.c
gpuarray_buffer_cuda.c
gpuarray_blas_cuda_cublas.c
gpuarray_collectives_cuda_nccl.c
gpuarray_buffer_opencl.c
gpuarray_blas_opencl_clblas.c
gpuarray_blas_opencl_clblast.c
)
set_property(SOURCE gpuarray_buffer_cuda.c APPEND PROPERTY OBJECT_DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/cluda_cuda.h.c)
set_property(SOURCE gpuarray_buffer_opencl.c APPEND PROPERTY OBJECT_DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/cluda_opencl.h.c)
check_function_exists(strlcat HAVE_STRL)
check_function_exists(mkstemp HAVE_MKSTEMP)
if(UNIX)
add_definitions(-D_GNU_SOURCE)
endif()
if(NOT HAVE_STRL)
list(APPEND _GPUARRAY_SRC gpuarray_strl.c)
endif()
if(NOT HAVE_MKSTEMP)
list(APPEND _GPUARRAY_SRC gpuarray_mkstemp.c)
endif()
configure_file(
${CMAKE_CURRENT_SOURCE_DIR}/private_config.h.in
${CMAKE_CURRENT_SOURCE_DIR}/private_config.h
)
add_subdirectory(util)
add_subdirectory(loaders)
set_rel(GPUARRAY_SRC ${_GPUARRAY_SRC})
list(APPEND GPUARRAY_SRC ${UTIL_SRC} ${LOADERS_SRC})
add_library(gpuarray SHARED ${GPUARRAY_SRC})
set_target_properties(gpuarray PROPERTIES
COMPILE_FLAGS "-DGPUARRAY_BUILDING_DLL -DGPUARRAY_SHARED"
INSTALL_NAME_DIR ${CMAKE_INSTALL_PREFIX}/lib
MACOSX_RPATH OFF
# This is the shared library version
VERSION 3.0
)
add_library(gpuarray-static STATIC ${GPUARRAY_SRC})
target_link_libraries(gpuarray ${CMAKE_DL_LIBS})
target_link_libraries(gpuarray-static ${CMAKE_DL_LIBS})
# Generate gpuarray/abi_version.h that contains the ABI version number.
get_target_property(GPUARRAY_ABI_VERSION gpuarray VERSION)
string(REPLACE "." ";" GPUARRAY_ABI_VERSION_NUMBERS ${GPUARRAY_ABI_VERSION})
list(GET GPUARRAY_ABI_VERSION_NUMBERS 0 GPUARRAY_ABI_VERSION_MAJOR)
list(GET GPUARRAY_ABI_VERSION_NUMBERS 1 GPUARRAY_ABI_VERSION_MINOR)
math(EXPR GPUARRAY_ABI_NUMBER "1000*${GPUARRAY_ABI_VERSION_MAJOR} + ${GPUARRAY_ABI_VERSION_MINOR}")
FILE(WRITE gpuarray/abi_version.h
"\#ifndef GPUARRAY_ABI_VERSION\n\#define GPUARRAY_ABI_VERSION ${GPUARRAY_ABI_NUMBER}\n\#endif\n"
)
# set SOVERSION and ensure it is the first part of VERSION.
set_property(TARGET gpuarray PROPERTY SOVERSION ${GPUARRAY_ABI_VERSION_MAJOR})
set(headers
gpuarray/array.h
gpuarray/blas.h
gpuarray/collectives.h
gpuarray/buffer.h
gpuarray/buffer_blas.h
gpuarray/buffer_collectives.h
gpuarray/abi_version.h
gpuarray/config.h
gpuarray/elemwise.h
gpuarray/error.h
gpuarray/extension.h
gpuarray/ext_cuda.h
gpuarray/kernel.h
gpuarray/types.h
gpuarray/util.h
)
install(FILES ${headers} DESTINATION include/gpuarray)
if(NOT UNIX)
install(FILES gpuarray/wincompat/stdint.h DESTINATION include/gpuarray/wincompat)
endif()
install(TARGETS gpuarray gpuarray-static
RUNTIME DESTINATION bin
LIBRARY DESTINATION lib
ARCHIVE DESTINATION lib
)
|
