@@ -41,12 +41,26 @@ function(build_umf_test)
4141
4242 set (LIB_DIRS ${LIB_DIRS} ${LIBHWLOC_LIBRARY_DIRS} )
4343
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)
4554 set (LIB_DIRS ${LIB_DIRS} ${JEMALLOC_LIBRARY_DIRS} )
55+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_JEMALLOC_ENABLED=1)
4656 endif ()
4757
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)
5064 endif ()
5165
5266 set (TEST_LIBS
@@ -61,15 +75,7 @@ function(build_umf_test)
6175 SRCS ${ARG_SRCS}
6276 LIBS ${TEST_LIBS} )
6377
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} )
7379
7480 if (NOT MSVC )
7581 # Suppress 'cast discards const qualifier' warnings. Parametrized GTEST
@@ -81,6 +87,7 @@ function(build_umf_test)
8187 target_compile_options (${TEST_TARGET_NAME} PRIVATE -Werror)
8288 endif ()
8389 endif ()
90+
8491 target_link_directories (${TEST_TARGET_NAME} PRIVATE ${LIB_DIRS} )
8592
8693 target_include_directories (
@@ -91,7 +98,8 @@ function(build_umf_test)
9198 ${UMF_CMAKE_SOURCE_DIR} /src/coarse
9299 ${UMF_CMAKE_SOURCE_DIR} /src/utils
93100 ${UMF_TEST_DIR} /common
94- ${UMF_TEST_DIR} )
101+ ${UMF_TEST_DIR}
102+ ${INC_DIRS} )
95103endfunction ()
96104
97105function (add_umf_test)
@@ -161,6 +169,10 @@ if(UMF_POOL_JEMALLOC_ENABLED)
161169 set (LIB_JEMALLOC_POOL jemalloc_pool)
162170endif ()
163171
172+ if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
173+ set (LIB_DISJOINT_POOL disjoint_pool)
174+ endif ()
175+
164176if (UMF_BUILD_SHARED_LIBRARY)
165177 # if build as shared library, ba symbols won't be visible in tests
166178 set (BA_SOURCES_FOR_TEST ${BA_SOURCES} )
@@ -254,13 +266,7 @@ if(LINUX AND (NOT UMF_DISABLE_HWLOC)) # OS-specific functions are implemented
254266 add_umf_test(
255267 NAME provider_os_memory
256268 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} )
264270 add_umf_test(
265271 NAME provider_os_memory_multiple_numa_nodes
266272 SRCS provider_os_memory_multiple_numa_nodes.cpp
@@ -374,8 +380,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
374380 SRCS providers/provider_level_zero.cpp
375381 ${UMF_UTILS_DIR} /utils_level_zero.cpp ${BA_SOURCES_FOR_TEST}
376382 LIBS ${UMF_UTILS_FOR_TEST} ze_loader)
377- target_include_directories (umf_test-provider_level_zero
378- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
379383
380384 add_umf_test(
381385 NAME provider_level_zero_dlopen
@@ -384,8 +388,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
384388 LIBS ${UMF_UTILS_FOR_TEST} )
385389 target_compile_definitions (umf_test-provider_level_zero_dlopen
386390 PUBLIC USE_DLOPEN=1)
387- target_include_directories (umf_test-provider_level_zero_dlopen
388- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
389391endif ()
390392
391393if (NOT UMF_BUILD_LEVEL_ZERO_PROVIDER)
@@ -405,10 +407,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
405407 SRCS providers/provider_cuda.cpp providers/cuda_helpers.cpp
406408 ${BA_SOURCES_FOR_TEST}
407409 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} )
412410
413411 add_umf_test(
414412 NAME provider_cuda_dlopen
@@ -417,8 +415,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
417415 LIBS ${UMF_UTILS_FOR_TEST} )
418416 target_compile_definitions (umf_test-provider_cuda_dlopen
419417 PUBLIC USE_DLOPEN=1)
420- target_include_directories (umf_test-provider_cuda_dlopen
421- PRIVATE ${CUDA_INCLUDE_DIRS} )
422418 else ()
423419 message (
424420 STATUS
@@ -610,10 +606,6 @@ if(LINUX)
610606 ze_loader
611607 disjoint_pool
612608 ${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} )
617609 add_umf_ipc_test(TEST ipc_level_zero_prov SRC_DIR providers)
618610 endif ()
619611
@@ -644,10 +636,6 @@ if(LINUX)
644636 cuda
645637 disjoint_pool
646638 ${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} )
651639 add_umf_ipc_test(TEST ipc_cuda_prov SRC_DIR providers)
652640 endif ()
653641else ()
0 commit comments