@@ -41,12 +41,26 @@ function(build_umf_test)
41
41
42
42
set (LIB_DIRS ${LIB_DIRS} ${LIBHWLOC_LIBRARY_DIRS} )
43
43
44
- if (UMF_BUILD_LIBUMF_POOL_JEMALLOC )
44
+ if (UMF_BUILD_CUDA_PROVIDER )
45
+ set (INC_DIRS ${INC_DIRS} ${CUDA_INCLUDE_DIRS} )
46
+ set (LIB_DIRS ${LIB_DIRS} ${CUDA_LIBRARY_DIRS} )
47
+ endif ()
48
+
49
+ if (UMF_BUILD_LEVEL_ZERO_PROVIDER )
50
+ set (INC_DIRS ${INC_DIRS} ${LEVEL_ZERO_INCLUDE_DIRS} )
51
+ endif ()
52
+
53
+ if (UMF_POOL_JEMALLOC_ENABLED )
45
54
set (LIB_DIRS ${LIB_DIRS} ${JEMALLOC_LIBRARY_DIRS} )
55
+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_JEMALLOC_ENABLED=1 )
46
56
endif ()
47
57
48
- if (UMF_BUILD_CUDA_PROVIDER )
49
- set (LIB_DIRS ${LIB_DIRS} ${CUDA_LIBRARY_DIRS} )
58
+ if (UMF_POOL_SCALABLE_ENABLED )
59
+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_SCALABLE_ENABLED=1 )
60
+ endif ()
61
+
62
+ if (UMF_BUILD_LIBUMF_POOL_DISJOINT )
63
+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_DISJOINT_ENABLED=1 )
50
64
endif ()
51
65
52
66
set (TEST_LIBS
@@ -61,15 +75,7 @@ function(build_umf_test)
61
75
SRCS ${ARG_SRCS}
62
76
LIBS ${TEST_LIBS} )
63
77
64
- if (UMF_POOL_JEMALLOC_ENABLED )
65
- target_compile_definitions (${TEST_TARGET_NAME}
66
- PRIVATE UMF_POOL_JEMALLOC_ENABLED=1 )
67
- endif ()
68
-
69
- if (UMF_POOL_SCALABLE_ENABLED )
70
- target_compile_definitions (${TEST_TARGET_NAME}
71
- PRIVATE UMF_POOL_SCALABLE_ENABLED=1 )
72
- endif ()
78
+ target_compile_definitions (${TEST_TARGET_NAME} PRIVATE ${CPL_DEFS} )
73
79
74
80
if (NOT MSVC )
75
81
# Suppress 'cast discards const qualifier' warnings. Parametrized GTEST
@@ -81,6 +87,7 @@ function(build_umf_test)
81
87
target_compile_options (${TEST_TARGET_NAME} PRIVATE -Werror )
82
88
endif ()
83
89
endif ()
90
+
84
91
target_link_directories (${TEST_TARGET_NAME} PRIVATE ${LIB_DIRS} )
85
92
86
93
target_include_directories (
@@ -91,7 +98,8 @@ function(build_umf_test)
91
98
${UMF_CMAKE_SOURCE_DIR} /src/coarse
92
99
${UMF_CMAKE_SOURCE_DIR} /src/utils
93
100
${UMF_TEST_DIR} /common
94
- ${UMF_TEST_DIR} )
101
+ ${UMF_TEST_DIR}
102
+ ${INC_DIRS} )
95
103
endfunction ()
96
104
97
105
function (add_umf_test )
@@ -161,6 +169,10 @@ if(UMF_POOL_JEMALLOC_ENABLED)
161
169
set (LIB_JEMALLOC_POOL jemalloc_pool )
162
170
endif ()
163
171
172
+ if (UMF_BUILD_LIBUMF_POOL_DISJOINT )
173
+ set (LIB_DISJOINT_POOL disjoint_pool )
174
+ endif ()
175
+
164
176
if (UMF_BUILD_SHARED_LIBRARY )
165
177
# if build as shared library, ba symbols won't be visible in tests
166
178
set (BA_SOURCES_FOR_TEST ${BA_SOURCES} )
@@ -254,13 +266,7 @@ if(LINUX AND (NOT UMF_DISABLE_HWLOC)) # OS-specific functions are implemented
254
266
add_umf_test (
255
267
NAME provider_os_memory
256
268
SRCS provider_os_memory.cpp ${BA_SOURCES_FOR_TEST}
257
- LIBS ${UMF_UTILS_FOR_TEST} ${LIB_JEMALLOC_POOL} )
258
- if (UMF_BUILD_LIBUMF_POOL_DISJOINT )
259
- target_compile_definitions (umf_test-provider_os_memory
260
- PRIVATE UMF_POOL_DISJOINT_ENABLED=1 )
261
- target_link_libraries (umf_test-provider_os_memory PRIVATE disjoint_pool )
262
- endif ()
263
-
269
+ LIBS ${UMF_UTILS_FOR_TEST} ${LIB_JEMALLOC_POOL} ${LIB_DISJOINT_POOL} )
264
270
add_umf_test (
265
271
NAME provider_os_memory_multiple_numa_nodes
266
272
SRCS provider_os_memory_multiple_numa_nodes.cpp
@@ -374,8 +380,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
374
380
SRCS providers/provider_level_zero.cpp
375
381
${UMF_UTILS_DIR} /utils_level_zero.cpp ${BA_SOURCES_FOR_TEST}
376
382
LIBS ${UMF_UTILS_FOR_TEST} ze_loader )
377
- target_include_directories (umf_test-provider_level_zero
378
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
379
383
380
384
add_umf_test (
381
385
NAME provider_level_zero_dlopen
@@ -384,8 +388,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
384
388
LIBS ${UMF_UTILS_FOR_TEST} )
385
389
target_compile_definitions (umf_test-provider_level_zero_dlopen
386
390
PUBLIC USE_DLOPEN=1 )
387
- target_include_directories (umf_test-provider_level_zero_dlopen
388
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
389
391
endif ()
390
392
391
393
if (NOT UMF_BUILD_LEVEL_ZERO_PROVIDER )
@@ -405,10 +407,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
405
407
SRCS providers/provider_cuda.cpp providers/cuda_helpers.cpp
406
408
${BA_SOURCES_FOR_TEST}
407
409
LIBS ${UMF_UTILS_FOR_TEST} cuda )
408
- target_include_directories (umf_test-provider_cuda
409
- PRIVATE ${CUDA_INCLUDE_DIRS} )
410
- target_link_directories (umf_test-provider_cuda PRIVATE
411
- ${CUDA_LIBRARY_DIRS} )
412
410
413
411
add_umf_test (
414
412
NAME provider_cuda_dlopen
@@ -417,8 +415,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
417
415
LIBS ${UMF_UTILS_FOR_TEST} )
418
416
target_compile_definitions (umf_test-provider_cuda_dlopen
419
417
PUBLIC USE_DLOPEN=1 )
420
- target_include_directories (umf_test-provider_cuda_dlopen
421
- PRIVATE ${CUDA_INCLUDE_DIRS} )
422
418
else ()
423
419
message (
424
420
STATUS
@@ -610,10 +606,6 @@ if(LINUX)
610
606
ze_loader
611
607
disjoint_pool
612
608
${UMF_UTILS_FOR_TEST} )
613
- target_include_directories (umf_test-ipc_level_zero_prov_producer
614
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
615
- target_include_directories (umf_test-ipc_level_zero_prov_consumer
616
- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
617
609
add_umf_ipc_test (TEST ipc_level_zero_prov SRC_DIR providers )
618
610
endif ()
619
611
@@ -644,10 +636,6 @@ if(LINUX)
644
636
cuda
645
637
disjoint_pool
646
638
${UMF_UTILS_FOR_TEST} )
647
- target_include_directories (umf_test-ipc_cuda_prov_producer
648
- PRIVATE ${CUDA_INCLUDE_DIRS} )
649
- target_include_directories (umf_test-ipc_cuda_prov_consumer
650
- PRIVATE ${CUDA_INCLUDE_DIRS} )
651
639
add_umf_ipc_test (TEST ipc_cuda_prov SRC_DIR providers )
652
640
endif ()
653
641
else ()
0 commit comments