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(CheckCSourceCompiles)
include(CheckLibraryExists)
find_package(PkgConfig)
pkg_search_module(CHECK check)
if(CHECK_FOUND)
if(CHECK_VERSION VERSION_LESS 0.10.0)
MESSAGE( "Check version older than 0.10.0" )
set(CHECK_FOUND 0)
endif()
else()
find_path(CHECK_INCLUDE_DIRS check.h)
find_library(CHECK_LIBRARIES NAMES check)
if(CHECK_INCLUDE_DIRS AND CHECK_LIBRARIES)
set(CHECK_CFLAGS)
set(CHECK_LIBRARY_DIRS)
set(CHECK_FOUND 1)
endif()
if(CHECK_FOUND)
set(CMAKE_REQUIRED_FLAGS ${CHECK_C_FLAGS} ${CHECK_LDFLAGS_OTHERS})
set(CMAKE_REQUIRED_INCLUDES ${CHECK_INCLUDE_DIRS})
CHECK_LIBRARY_EXISTS(pthread pthread_create "" HAVE_PTHREAD)
if (HAVE_PTHREAD)
set(CHECK_LIBRARIES ${CHECK_LIBRARIES} pthread)
endif (HAVE_PTHREAD)
CHECK_LIBRARY_EXISTS(rt nanosleep "" HAVE_LIBRT)
if (HAVE_LIBRT)
set(CHECK_LIBRARIES ${CHECK_LIBRARIES} rt)
endif (HAVE_LIBRT)
CHECK_LIBRARY_EXISTS(m cos "" HAVE_LIBM)
if (HAVE_LIBM)
set(CHECK_LIBRARIES ${CHECK_LIBRARIES} m)
endif (HAVE_LIBM)
set(CMAKE_REQUIRED_LIBRARIES ${CHECK_LIBRARIES})
CHECK_C_SOURCE_COMPILES(
"#include <check.h>
int main() {
ck_assert_ptr_ne(NULL, NULL);
}"
CHECK_FUNCS)
if (NOT CHECK_FUNCS)
set(CHECK_FOUND 0)
endif()
endif()
endif()
if(CHECK_FOUND)
enable_testing()
include_directories("${CMAKE_SOURCE_DIR}/src")
include_directories("${CMAKE_CURRENT_SOURCE_DIR}")
include_directories(${CHECK_INCLUDE_DIRS})
link_directories(${CHECK_LIBRARY_DIRS})
foreach(flag ${CHECK_C_FLAGS})
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${flag}")
endforeach()
foreach(flag ${CHECK_LDFLAGS_OTHER})
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${flag}")
endforeach()
add_executable(check_types main.c check_types.c)
target_link_libraries(check_types ${CHECK_LIBRARIES} gpuarray)
add_test(test_types "${CMAKE_CURRENT_BINARY_DIR}/check_types")
add_executable(check_util main.c check_util.c)
target_link_libraries(check_util ${CHECK_LIBRARIES} gpuarray)
add_test(test_util "${CMAKE_CURRENT_BINARY_DIR}/check_util")
add_executable(check_util_integerfactoring main.c check_util_integerfactoring.c)
target_link_libraries(check_util_integerfactoring ${CHECK_LIBRARIES} gpuarray-static)
add_test(test_util_integerfactoring "${CMAKE_CURRENT_BINARY_DIR}/check_util_integerfactoring")
add_executable(check_reduction main.c device.c check_reduction.c)
target_link_libraries(check_reduction ${CHECK_LIBRARIES} gpuarray)
add_test(test_reduction "${CMAKE_CURRENT_BINARY_DIR}/check_reduction")
add_executable(check_array main.c device.c check_array.c)
target_link_libraries(check_array ${CHECK_LIBRARIES} gpuarray)
add_test(test_array "${CMAKE_CURRENT_BINARY_DIR}/check_array")
add_executable(check_blas main.c device.c check_blas.c)
target_link_libraries(check_blas ${CHECK_LIBRARIES} gpuarray)
add_test(test_blas "${CMAKE_CURRENT_BINARY_DIR}/check_blas")
add_executable(check_elemwise main.c device.c check_elemwise.c)
target_link_libraries(check_elemwise ${CHECK_LIBRARIES} gpuarray)
add_test(test_elemwise "${CMAKE_CURRENT_BINARY_DIR}/check_elemwise")
add_executable(check_error main.c check_error.c)
target_link_libraries(check_error ${CHECK_LIBRARIES} gpuarray)
add_test(test_error "${CMAKE_CURRENT_BINARY_DIR}/check_error")
add_executable(check_buffer main.c device.c check_buffer.c)
target_link_libraries(check_buffer ${CHECK_LIBRARIES} gpuarray)
add_test(test_buffer "${CMAKE_CURRENT_BINARY_DIR}/check_buffer")
find_package(MPI)
if (MPI_C_FOUND)
add_executable(check_buffer_collectives
main.c device.c communicator.c check_buffer_collectives.c
)
target_link_libraries(check_buffer_collectives
${CHECK_LIBRARIES} ${MPI_C_LIBRARIES} gpuarray
)
target_include_directories(check_buffer_collectives
PRIVATE ${MPI_C_INCLUDE_PATH}
)
add_executable(check_collectives
main.c device.c communicator.c check_collectives.c
)
target_link_libraries(check_collectives
${CHECK_LIBRARIES} ${MPI_C_LIBRARIES} gpuarray
)
target_include_directories(check_collectives
PRIVATE ${MPI_C_INCLUDE_PATH}
)
set_target_properties(check_buffer_collectives check_collectives PROPERTIES
COMPILE_DEFINITIONS TEST_COLLECTIVES
COMPILE_FLAGS "${MPI_C_COMPILE_FLAGS}"
LINK_FLAGS "${MPI_C_LINK_FLAGS}"
)
set(_NUM_DEVS $ENV{NUM_DEVS})
if(NOT _NUM_DEVS)
set(_NUM_DEVS 1)
endif()
set(_DEV_NAMES $ENV{DEV_NAMES})
if(NOT _DEV_NAMES)
set(_DEV_NAMES "cuda")
endif()
separate_arguments(_DEV_NAMES)
add_test(NAME test_buffer_collectives
COMMAND "${MPIEXEC}" ${MPIEXEC_NUMPROC_FLAG} ${_NUM_DEVS} ${MPIEXEC_PREFLAGS}
"${CMAKE_CURRENT_BINARY_DIR}/check_buffer_collectives" ${MPIEXEC_POSTFLAGS} ${_DEV_NAMES})
add_test(NAME test_collectives
COMMAND ${MPIEXEC} ${MPIEXEC_NUMPROC_FLAG} ${_NUM_DEVS} ${MPIEXEC_PREFLAGS}
"${CMAKE_CURRENT_BINARY_DIR}/check_collectives" ${MPIEXEC_POSTFLAGS} ${_DEV_NAMES})
else()
message(WARNING "Cannot find MPI")
message(WARNING "Checks on collectives and buffer_collectives will not be built or performed.")
endif()
ELSE(CHECK_FOUND)
MESSAGE("Tests disabled because Check was not found")
ENDIF(CHECK_FOUND)
|
