Skip to content

Commit 6e3a84c

Browse files
ldorauldorau
committed
Add add_umf_benchmark() to remove duplicated code in benchmarks CMakeLists
Signed-off-by: Lukasz Dorau <[email protected]>
1 parent 69cdec6 commit 6e3a84c

File tree

1 file changed

+75
-65
lines changed

1 file changed

+75
-65
lines changed

‎benchmark/CMakeLists.txt

Lines changed: 75 additions & 65 deletions
Original file line numberDiff line numberDiff line change
@@ -2,92 +2,102 @@
22
# Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
33
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
44

5-
include(FindThreads)
6-
75
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
86
message(WARNING "The benchmarks SHOULD NOT be run in the Debug build type!")
97
endif()
108

9+
if (UMF_BUILD_BENCHMARKS_MT)
10+
include(FindThreads)
11+
endif()
12+
13+
function(add_umf_benchmark)
14+
# NAME - a name of the benchmark
15+
# SRCS - source files
16+
# LIBS - libraries to be linked with
17+
# LIBDIRS - directories of libraries to be linked with
18+
set(oneValueArgs NAME)
19+
set(multiValueArgs SRCS LIBS LIBDIRS)
20+
cmake_parse_arguments(ARG "" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
21+
22+
set(BENCH_NAME umf-bench-${ARG_NAME})
23+
24+
set(BENCH_LIBS umf ${ARG_LIBS})
25+
26+
add_umf_executable(NAME ${BENCH_NAME} SRCS ${ARG_SRCS} LIBS ${BENCH_LIBS})
27+
28+
target_include_directories(${BENCH_NAME} PRIVATE
29+
${UMF_CMAKE_SOURCE_DIR}/include
30+
${UMF_CMAKE_SOURCE_DIR}/src/utils)
31+
32+
target_link_directories(${BENCH_NAME} PRIVATE ${ARG_LIBDIRS})
33+
34+
add_test(NAME ${BENCH_NAME}
35+
COMMAND ${BENCH_NAME}
36+
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
37+
38+
# Benchmark passes if it prints "PASSED" in the output,
39+
# because ubench of scalable pool fails if the confidence interval
40+
# exceeds maximum permitted 2.5%.
41+
set_tests_properties(${BENCH_NAME} PROPERTIES
42+
LABELS "benchmark"
43+
PASS_REGULAR_EXPRESSION "PASSED")
44+
45+
if(WINDOWS)
46+
# append PATH to DLLs
47+
set(DLL_PATH_LIST "PATH=path_list_append:../bin/$<CONFIG>")
48+
if(LIBHWLOC_LIBRARY_DIRS)
49+
set(DLL_PATH_LIST "${DLL_PATH_LIST};PATH=path_list_append:${LIBHWLOC_LIBRARY_DIRS}/../bin")
50+
endif()
51+
if(TBB_LIBRARY_DIRS)
52+
set(DLL_PATH_LIST "${DLL_PATH_LIST};PATH=path_list_append:${TBB_LIBRARY_DIRS}/../bin")
53+
endif()
54+
set_property(TEST ${EXAMPLE_NAME} PROPERTY ENVIRONMENT_MODIFICATION "${DLL_PATH_LIST}")
55+
endif()
56+
57+
if (UMF_BUILD_OS_MEMORY_PROVIDER)
58+
target_compile_definitions(${BENCH_NAME} PRIVATE UMF_BUILD_OS_MEMORY_PROVIDER=1)
59+
endif()
60+
if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
61+
target_compile_definitions(${BENCH_NAME} PRIVATE UMF_BUILD_LIBUMF_POOL_DISJOINT=1)
62+
endif()
63+
if (UMF_BUILD_LIBUMF_POOL_JEMALLOC)
64+
target_compile_definitions(${BENCH_NAME} PRIVATE UMF_BUILD_LIBUMF_POOL_JEMALLOC=1)
65+
endif()
66+
if (UMF_BUILD_LIBUMF_POOL_SCALABLE)
67+
target_compile_definitions(${BENCH_NAME} PRIVATE UMF_BUILD_LIBUMF_POOL_SCALABLE=1)
68+
endif()
69+
endfunction()
70+
71+
# optional libraries
1172
if(UMF_BUILD_LIBUMF_POOL_DISJOINT)
1273
set(LIBS_OPTIONAL ${LIBS_OPTIONAL} disjoint_pool)
1374
endif()
14-
1575
if(UMF_BUILD_LIBUMF_POOL_JEMALLOC)
1676
set(LIBS_OPTIONAL ${LIBS_OPTIONAL} jemalloc_pool)
1777
endif()
18-
1978
if(UMF_BUILD_LIBUMF_POOL_SCALABLE)
2079
set(LIBS_OPTIONAL ${LIBS_OPTIONAL} scalable_pool)
2180
endif()
22-
23-
add_executable(ubench ubench.c)
24-
2581
if(LINUX)
26-
set(LIBS_LINUX m)
82+
set(LIBS_OPTIONAL ${LIBS_OPTIONAL} m)
2783
endif()
28-
29-
add_dependencies(ubench
30-
umf
31-
${LIBS_OPTIONAL})
32-
3384
if(UMF_BUILD_OS_MEMORY_PROVIDER)
3485
set(LIBS_OPTIONAL ${LIBS_OPTIONAL} ${LIBHWLOC_LIBRARIES})
3586
set(LIB_DIRS_OPTIONAL ${LIB_DIRS_OPTIONAL} ${LIBHWLOC_LIBRARY_DIRS})
3687
endif()
3788

38-
target_include_directories(ubench PRIVATE ${UMF_CMAKE_SOURCE_DIR}/include/ ${UMF_CMAKE_SOURCE_DIR}/src/utils/)
89+
# BENCHMARKS
3990

40-
target_link_directories(ubench PRIVATE ${LIB_DIRS_OPTIONAL})
41-
42-
target_link_libraries(ubench
43-
umf
44-
${LIBS_OPTIONAL}
45-
${CMAKE_THREAD_LIBS_INIT}
46-
${LIBS_LINUX})
91+
add_umf_benchmark(
92+
NAME ubench
93+
SRCS ubench.c
94+
LIBS ${LIBS_OPTIONAL}
95+
LIBDIRS ${LIB_DIRS_OPTIONAL})
4796

4897
if (UMF_BUILD_BENCHMARKS_MT)
49-
add_executable(multithread_bench multithread.cpp)
50-
target_link_directories(multithread_bench PRIVATE ${LIB_DIRS_OPTIONAL})
51-
target_link_libraries(multithread_bench
52-
umf
53-
${LIBS_OPTIONAL}
54-
${CMAKE_THREAD_LIBS_INIT}
55-
${LIBS_LINUX})
56-
target_include_directories(multithread_bench PRIVATE ${UMF_CMAKE_SOURCE_DIR}/include/)
57-
endif()
58-
59-
if (UMF_BUILD_OS_MEMORY_PROVIDER)
60-
target_compile_definitions(ubench PRIVATE UMF_BUILD_OS_MEMORY_PROVIDER=1)
61-
62-
if (UMF_BUILD_BENCHMARKS_MT)
63-
target_compile_definitions(multithread_bench PRIVATE UMF_BUILD_OS_MEMORY_PROVIDER=1)
64-
endif()
65-
endif()
66-
67-
if (UMF_BUILD_OS_MEMORY_PROVIDER)
68-
target_compile_definitions(ubench PRIVATE UMF_BUILD_OS_MEMORY_PROVIDER=1)
69-
endif()
70-
71-
if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
72-
target_compile_definitions(ubench PRIVATE UMF_BUILD_LIBUMF_POOL_DISJOINT=1)
73-
74-
if (UMF_BUILD_BENCHMARKS_MT)
75-
target_compile_definitions(multithread_bench PRIVATE UMF_BUILD_LIBUMF_POOL_DISJOINT=1)
76-
endif()
77-
endif()
78-
79-
if (UMF_BUILD_LIBUMF_POOL_JEMALLOC)
80-
target_compile_definitions(ubench PRIVATE UMF_BUILD_LIBUMF_POOL_JEMALLOC=1)
81-
82-
if (UMF_BUILD_BENCHMARKS_MT)
83-
target_compile_definitions(multithread_bench PRIVATE UMF_BUILD_LIBUMF_POOL_JEMALLOC=1)
84-
endif()
85-
endif()
86-
87-
if (UMF_BUILD_LIBUMF_POOL_SCALABLE)
88-
target_compile_definitions(ubench PRIVATE UMF_BUILD_LIBUMF_POOL_SCALABLE=1)
89-
90-
if (UMF_BUILD_BENCHMARKS_MT)
91-
target_compile_definitions(multithread_bench PRIVATE UMF_BUILD_LIBUMF_POOL_SCALABLE=1)
92-
endif()
98+
add_umf_benchmark(
99+
NAME multithreaded
100+
SRCS multithread.cpp
101+
LIBS ${LIBS_OPTIONAL} ${CMAKE_THREAD_LIBS_INIT}
102+
LIBDIRS ${LIB_DIRS_OPTIONAL})
93103
endif()

0 commit comments

Comments
 (0)