Add looking for the tbbmalloc library and the header

Add looking for the tbbmalloc library
and the <tbb/scalable_allocator.h> header.

Signed-off-by: Lukasz Dorau <[email protected]>

Author: ldorau

CMakeLists.txt

@@ -117,6 +117,11 @@ if(UMF_BUILD_OS_MEMORY_PROVIDER AND (LINUX OR WINDOWS))
     find_package(LIBHWLOC REQUIRED hwloc)
 endif()
 
+if(UMF_BUILD_LIBUMF_POOL_SCALABLE AND (LINUX OR WINDOWS))
+    # Do not use pkgconfig, because it sets wrong paths on Windows
+    find_package(LIBTBB REQUIRED tbb)
+endif()
+
 add_subdirectory(src)
 
 if(UMF_BUILD_TESTS)

cmake/FindLIBTBB.cmake

@@ -0,0 +1,44 @@
+# Copyright (C) 2024 Intel Corporation
+# Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+message(STATUS "Checking for module 'libtbb' using find_library()")
+
+find_library(LIBTBB_LIBRARY NAMES libtbbmalloc tbbmalloc)
+set(LIBTBB_LIBRARIES ${LIBTBB_LIBRARY})
+
+find_file(LIBTBB_HEADER NAMES scalable_allocator.h HINTS /usr/include/tbb ${CMAKE_PREFIX_PATH}/include/tbb)
+if(LIBTBB_HEADER)
+	get_filename_component(LIBTBB_INCLUDE_DIR_TBB ${LIBTBB_HEADER} DIRECTORY)
+	get_filename_component(LIBTBB_INCLUDE_DIR ${LIBTBB_INCLUDE_DIR_TBB} DIRECTORY)
+	set(LIBTBB_INCLUDE_DIRS ${LIBTBB_INCLUDE_DIR})
+else()
+	set(MSG_NOT_FOUND "<tbb/scalable_allocator.h> header NOT found (set CMAKE_PREFIX_PATH to point the location)")
+	if(LIBTBB_FIND_REQUIRED)
+		message(FATAL_ERROR ${MSG_NOT_FOUND})
+	else()
+		message(WARNING ${MSG_NOT_FOUND})
+	endif()
+endif()
+
+if(WINDOWS)
+	find_file(LIBTBB_DLL NAMES tbbmalloc.dll HINTS ${CMAKE_PREFIX_PATH}/bin)
+	get_filename_component(LIBTBB_DLL_DIR ${LIBTBB_DLL} DIRECTORY)
+	set(LIBTBB_DLL_DIRS ${LIBTBB_DLL_DIR})
+endif()
+
+if(LIBTBB_LIBRARY)
+	message(STATUS "  Found libtbb using find_library()")
+	message(STATUS "    LIBTBB_LIBRARIES = ${LIBTBB_LIBRARIES}")
+	message(STATUS "    LIBTBB_INCLUDE_DIRS = ${LIBTBB_INCLUDE_DIRS}")
+	if(WINDOWS)
+		message(STATUS "    LIBTBB_DLL_DIRS = ${LIBTBB_DLL_DIRS}")
+	endif()
+else()
+	set(MSG_NOT_FOUND "libtbb NOT found (set CMAKE_PREFIX_PATH to point the location)")
+	if(LIBTBB_FIND_REQUIRED)
+		message(FATAL_ERROR ${MSG_NOT_FOUND})
+	else()
+		message(WARNING ${MSG_NOT_FOUND})
+	endif()
+endif()