@@ -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