Add ba_alloc sources to pool libs

if umf is build as a shared lib. This is
needed because ba_alloc symbols are not longer
exported from libumf

Author: igchor

src/pool/CMakeLists.txt

@@ -2,11 +2,15 @@
 # Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
+if(UMF_BUILD_SHARED_LIBRARY)
+    set(POOL_EXTRA_SRCS ${BA_SOURCES})
+endif()
+
 # libumf_pool_disjoint
 if(UMF_BUILD_LIBUMF_POOL_DISJOINT)
     add_umf_library(NAME disjoint_pool
                     TYPE STATIC
-                    SRCS pool_disjoint.cpp
+                    SRCS pool_disjoint.cpp ${POOL_EXTRA_SRCS}
                     LIBS umf_utils)
 
     add_library(${PROJECT_NAME}::disjoint_pool ALIAS disjoint_pool)
@@ -34,7 +38,7 @@ if(UMF_BUILD_LIBUMF_POOL_JEMALLOC)
         # This is a temporary fix for https://github.com/oneapi-src/unified-memory-framework/issues/161
         add_umf_library(NAME jemalloc_pool
                         TYPE STATIC
-                        SRCS pool_jemalloc.c
+                        SRCS pool_jemalloc.c ${POOL_EXTRA_SRCS}
                         LIBS c jemalloc umf_utils)
         add_library(${PROJECT_NAME}::jemalloc_pool ALIAS jemalloc_pool)
         install(TARGETS jemalloc_pool
@@ -50,7 +54,7 @@ if(UMF_BUILD_LIBUMF_POOL_SCALABLE)
     if(LINUX)
         add_umf_library(NAME scalable_pool
                         TYPE STATIC
-                        SRCS pool_scalable.c
+                        SRCS pool_scalable.c ${POOL_EXTRA_SRCS}
                         LIBS dl umf_utils)
         add_library(${PROJECT_NAME}::scalable_pool ALIAS scalable_pool)
         install(TARGETS scalable_pool