@@ -40,12 +40,26 @@ function(build_umf_test)
40
40
41
41
set (LIB_DIRS ${LIB_DIRS} ${LIBHWLOC_LIBRARY_DIRS} )
42
42
43
- if (UMF_BUILD_LIBUMF_POOL_JEMALLOC)
43
+ if (UMF_BUILD_CUDA_PROVIDER)
44
+ set (INC_DIRS ${INC_DIRS} ${CUDA_INCLUDE_DIRS} )
45
+ set (LIB_DIRS ${LIB_DIRS} ${CUDA_LIBRARY_DIRS} )
46
+ endif ()
47
+
48
+ if (UMF_BUILD_LEVEL_ZERO_PROVIDER)
49
+ set (INC_DIRS ${INC_DIRS} ${LEVEL_ZERO_INCLUDE_DIRS} )
50
+ endif ()
51
+
52
+ if (UMF_POOL_JEMALLOC_ENABLED)
44
53
set (LIB_DIRS ${LIB_DIRS} ${JEMALLOC_LIBRARY_DIRS} )
54
+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_JEMALLOC_ENABLED=1)
45
55
endif ()
46
56
47
- if (UMF_BUILD_CUDA_PROVIDER)
48
- set (LIB_DIRS ${LIB_DIRS} ${CUDA_LIBRARY_DIRS} )
57
+ if (UMF_POOL_SCALABLE_ENABLED)
58
+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_SCALABLE_ENABLED=1)
59
+ endif ()
60
+
61
+ if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
62
+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_DISJOINT_ENABLED=1)
49
63
endif ()
50
64
51
65
set (TEST_LIBS
@@ -60,15 +74,7 @@ function(build_umf_test)
60
74
SRCS ${ARG_SRCS}
61
75
LIBS ${TEST_LIBS} )
62
76
63
- if (UMF_POOL_JEMALLOC_ENABLED)
64
- target_compile_definitions (${TEST_TARGET_NAME}
65
- PRIVATE UMF_POOL_JEMALLOC_ENABLED=1)
66
- endif ()
67
-
68
- if (UMF_POOL_SCALABLE_ENABLED)
69
- target_compile_definitions (${TEST_TARGET_NAME}
70
- PRIVATE UMF_POOL_SCALABLE_ENABLED=1)
71
- endif ()
77
+ target_compile_definitions (${TEST_TARGET_NAME} PRIVATE ${CPL_DEFS} )
72
78
73
79
if (NOT MSVC )
74
80
# Suppress 'cast discards const qualifier' warnings. Parametrized GTEST
@@ -80,6 +86,7 @@ function(build_umf_test)
80
86
target_compile_options (${TEST_TARGET_NAME} PRIVATE -Werror)
81
87
endif ()
82
88
endif ()
89
+
83
90
target_link_directories (${TEST_TARGET_NAME} PRIVATE ${LIB_DIRS} )
84
91
85
92
target_include_directories (
@@ -89,7 +96,8 @@ function(build_umf_test)
89
96
${UMF_CMAKE_SOURCE_DIR} /src/base_alloc
90
97
${UMF_CMAKE_SOURCE_DIR} /src/utils
91
98
${UMF_TEST_DIR} /common
92
- ${UMF_TEST_DIR} )
99
+ ${UMF_TEST_DIR}
100
+ ${INC_DIRS} )
93
101
endfunction ()
94
102
95
103
function (add_umf_test)
@@ -157,6 +165,10 @@ if(UMF_POOL_JEMALLOC_ENABLED)
157
165
set (LIB_JEMALLOC_POOL jemalloc_pool)
158
166
endif ()
159
167
168
+ if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
169
+ set (LIB_DISJOINT_POOL disjoint_pool)
170
+ endif ()
171
+
160
172
if (UMF_BUILD_SHARED_LIBRARY)
161
173
# if build as shared library, ba symbols won't be visible in tests
162
174
set (BA_SOURCES_FOR_TEST ${BA_SOURCES} )
@@ -245,13 +257,7 @@ if(LINUX AND (NOT UMF_DISABLE_HWLOC)) # OS-specific functions are implemented
245
257
add_umf_test(
246
258
NAME provider_os_memory
247
259
SRCS provider_os_memory.cpp ${BA_SOURCES_FOR_TEST}
248
- LIBS ${UMF_UTILS_FOR_TEST} ${LIB_JEMALLOC_POOL} )
249
- if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
250
- target_compile_definitions (umf_test-provider_os_memory
251
- PRIVATE UMF_POOL_DISJOINT_ENABLED=1)
252
- target_link_libraries (umf_test-provider_os_memory PRIVATE disjoint_pool)
253
- endif ()
254
-
260
+ LIBS ${UMF_UTILS_FOR_TEST} ${LIB_JEMALLOC_POOL} ${LIB_DISJOINT_POOL} )
255
261
add_umf_test(
256
262
NAME provider_os_memory_multiple_numa_nodes
257
263
SRCS provider_os_memory_multiple_numa_nodes.cpp
@@ -365,8 +371,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
365
371
SRCS providers/provider_level_zero.cpp providers/level_zero_helpers.cpp
366
372
${BA_SOURCES_FOR_TEST}
367
373
LIBS ${UMF_UTILS_FOR_TEST} ze_loader)
368
- target_include_directories (umf_test-provider_level_zero
369
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
370
374
371
375
add_umf_test(
372
376
NAME provider_level_zero_dlopen
@@ -375,8 +379,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
375
379
LIBS ${UMF_UTILS_FOR_TEST} )
376
380
target_compile_definitions (umf_test-provider_level_zero_dlopen
377
381
PUBLIC USE_DLOPEN=1)
378
- target_include_directories (umf_test-provider_level_zero_dlopen
379
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
380
382
endif ()
381
383
382
384
if (NOT UMF_BUILD_LEVEL_ZERO_PROVIDER)
@@ -396,10 +398,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
396
398
SRCS providers/provider_cuda.cpp providers/cuda_helpers.cpp
397
399
${BA_SOURCES_FOR_TEST}
398
400
LIBS ${UMF_UTILS_FOR_TEST} cuda)
399
- target_include_directories (umf_test-provider_cuda
400
- PRIVATE ${CUDA_INCLUDE_DIRS} )
401
- target_link_directories (umf_test-provider_cuda PRIVATE
402
- ${CUDA_LIBRARY_DIRS} )
403
401
404
402
add_umf_test(
405
403
NAME provider_cuda_dlopen
@@ -408,8 +406,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
408
406
LIBS ${UMF_UTILS_FOR_TEST} )
409
407
target_compile_definitions (umf_test-provider_cuda_dlopen
410
408
PUBLIC USE_DLOPEN=1)
411
- target_include_directories (umf_test-provider_cuda_dlopen
412
- PRIVATE ${CUDA_INCLUDE_DIRS} )
413
409
else ()
414
410
message (
415
411
STATUS
@@ -601,10 +597,6 @@ if(LINUX)
601
597
ze_loader
602
598
disjoint_pool
603
599
${UMF_UTILS_FOR_TEST} )
604
- target_include_directories (umf_test-ipc_level_zero_prov_producer
605
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
606
- target_include_directories (umf_test-ipc_level_zero_prov_consumer
607
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
608
600
add_umf_ipc_test(TEST ipc_level_zero_prov SRC_DIR providers)
609
601
endif ()
610
602
@@ -635,10 +627,6 @@ if(LINUX)
635
627
cuda
636
628
disjoint_pool
637
629
${UMF_UTILS_FOR_TEST} )
638
- target_include_directories (umf_test-ipc_cuda_prov_producer
639
- PRIVATE ${CUDA_INCLUDE_DIRS} )
640
- target_include_directories (umf_test-ipc_cuda_prov_consumer
641
- PRIVATE ${CUDA_INCLUDE_DIRS} )
642
630
add_umf_ipc_test(TEST ipc_cuda_prov SRC_DIR providers)
643
631
endif ()
644
632
else ()
0 commit comments