@@ -40,12 +40,26 @@ function(build_umf_test)
4040
4141 set (LIB_DIRS ${LIB_DIRS} ${LIBHWLOC_LIBRARY_DIRS} )
4242
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)
4453 set (LIB_DIRS ${LIB_DIRS} ${JEMALLOC_LIBRARY_DIRS} )
54+ set (CPL_DEFS ${CPL_DEFS} UMF_POOL_JEMALLOC_ENABLED=1)
4555 endif ()
4656
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)
4963 endif ()
5064
5165 set (TEST_LIBS
@@ -60,15 +74,7 @@ function(build_umf_test)
6074 SRCS ${ARG_SRCS}
6175 LIBS ${TEST_LIBS} )
6276
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} )
7278
7379 if (NOT MSVC )
7480 # Suppress 'cast discards const qualifier' warnings. Parametrized GTEST
@@ -80,6 +86,7 @@ function(build_umf_test)
8086 target_compile_options (${TEST_TARGET_NAME} PRIVATE -Werror)
8187 endif ()
8288 endif ()
89+
8390 target_link_directories (${TEST_TARGET_NAME} PRIVATE ${LIB_DIRS} )
8491
8592 target_include_directories (
@@ -89,7 +96,8 @@ function(build_umf_test)
8996 ${UMF_CMAKE_SOURCE_DIR} /src/base_alloc
9097 ${UMF_CMAKE_SOURCE_DIR} /src/utils
9198 ${UMF_TEST_DIR} /common
92- ${UMF_TEST_DIR} )
99+ ${UMF_TEST_DIR}
100+ ${INC_DIRS} )
93101endfunction ()
94102
95103function (add_umf_test)
@@ -157,6 +165,10 @@ if(UMF_POOL_JEMALLOC_ENABLED)
157165 set (LIB_JEMALLOC_POOL jemalloc_pool)
158166endif ()
159167
168+ if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
169+ set (LIB_DISJOINT_POOL disjoint_pool)
170+ endif ()
171+
160172if (UMF_BUILD_SHARED_LIBRARY)
161173 # if build as shared library, ba symbols won't be visible in tests
162174 set (BA_SOURCES_FOR_TEST ${BA_SOURCES} )
@@ -245,13 +257,7 @@ if(LINUX AND (NOT UMF_DISABLE_HWLOC)) # OS-specific functions are implemented
245257 add_umf_test(
246258 NAME provider_os_memory
247259 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} )
255261 add_umf_test(
256262 NAME provider_os_memory_multiple_numa_nodes
257263 SRCS provider_os_memory_multiple_numa_nodes.cpp
@@ -365,8 +371,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
365371 SRCS providers/provider_level_zero.cpp providers/level_zero_helpers.cpp
366372 ${BA_SOURCES_FOR_TEST}
367373 LIBS ${UMF_UTILS_FOR_TEST} ze_loader)
368- target_include_directories (umf_test-provider_level_zero
369- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
370374
371375 add_umf_test(
372376 NAME provider_level_zero_dlopen
@@ -375,8 +379,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
375379 LIBS ${UMF_UTILS_FOR_TEST} )
376380 target_compile_definitions (umf_test-provider_level_zero_dlopen
377381 PUBLIC USE_DLOPEN=1)
378- target_include_directories (umf_test-provider_level_zero_dlopen
379- PRIVATE ${LEVEL_ZERO_INCLUDE_DIRS} )
380382endif ()
381383
382384if (NOT UMF_BUILD_LEVEL_ZERO_PROVIDER)
@@ -396,10 +398,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
396398 SRCS providers/provider_cuda.cpp providers/cuda_helpers.cpp
397399 ${BA_SOURCES_FOR_TEST}
398400 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} )
403401
404402 add_umf_test(
405403 NAME provider_cuda_dlopen
@@ -408,8 +406,6 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_CUDA_PROVIDER)
408406 LIBS ${UMF_UTILS_FOR_TEST} )
409407 target_compile_definitions (umf_test-provider_cuda_dlopen
410408 PUBLIC USE_DLOPEN=1)
411- target_include_directories (umf_test-provider_cuda_dlopen
412- PRIVATE ${CUDA_INCLUDE_DIRS} )
413409 else ()
414410 message (
415411 STATUS
@@ -601,10 +597,6 @@ if(LINUX)
601597 ze_loader
602598 disjoint_pool
603599 ${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} )
608600 add_umf_ipc_test(TEST ipc_level_zero_prov SRC_DIR providers)
609601 endif ()
610602
@@ -635,10 +627,6 @@ if(LINUX)
635627 cuda
636628 disjoint_pool
637629 ${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} )
642630 add_umf_ipc_test(TEST ipc_cuda_prov SRC_DIR providers)
643631 endif ()
644632else ()
0 commit comments