Add FixedSizedHostBuffer

cmake/thirdparty/get_cucascade.cmake

@@ -29,6 +29,7 @@ function(find_and_configure_cucascade)
   rapids_cpm_find(
     cuCascade 0.1.0
     GLOBAL_TARGETS cuCascade::cucascade
+    BUILD_EXPORT_SET rapidsmpf-exports
     CPM_ARGS
     GIT_REPOSITORY https://github.com/NVIDIA/cuCascade.git
     GIT_TAG main
@@ -40,18 +41,6 @@ function(find_and_configure_cucascade)
             "CUCASCADE_WARNINGS_AS_ERRORS OFF"
     EXCLUDE_FROM_ALL
   )
-
-  # Create an interface library that wraps cuCascade to avoid export conflicts This target won't be
-  # exported but can be used internally. Link kvikio explicitly to satisfy cuDF's dependency.
-  if(TARGET cuCascade::cucascade AND NOT TARGET rapidsmpf_cucascade_internal)
-    add_library(rapidsmpf_cucascade_internal INTERFACE)
-    target_link_libraries(rapidsmpf_cucascade_internal INTERFACE cuCascade::cucascade)
-    # Link kvikio to ensure cuDF's transitive dependency is satisfied
-    if(TARGET kvikio::kvikio)
-      target_link_libraries(rapidsmpf_cucascade_internal INTERFACE kvikio::kvikio)
-    endif()
-    set_target_properties(rapidsmpf_cucascade_internal PROPERTIES EXPORT_NAME "")
-  endif()
 endfunction()
 
 find_and_configure_cucascade()

cpp/CMakeLists.txt

@@ -196,6 +196,7 @@ add_library(
   src/integrations/cudf/utils.cpp
   src/memory/buffer.cpp
   src/memory/buffer_resource.cpp
+  src/memory/fixed_sized_host_buffer.cpp
   src/memory/host_buffer.cpp
   src/memory/host_memory_resource.cpp
   src/memory/memory_reservation.cpp
@@ -311,10 +312,8 @@ endif()
 target_link_libraries(
   rapidsmpf
   PUBLIC rmm::rmm cudf::cudf CCCL::CCCL $<TARGET_NAME_IF_EXISTS:ucxx::ucxx>
-         $<TARGET_NAME_IF_EXISTS:libcoro>
+         $<TARGET_NAME_IF_EXISTS:libcoro> $<TARGET_NAME_IF_EXISTS:cuCascade::cucascade>
   PRIVATE cuco::cuco
-          $<TARGET_NAME_IF_EXISTS:rapidsmpf_cucascade_internal>
-          $<$<NOT:$<TARGET_EXISTS:rapidsmpf_cucascade_internal>>:cuCascade::cucascade>
           $<$<BOOL:${RAPIDSMPF_HAVE_NUMA}>:numa>
           $<TARGET_NAME_IF_EXISTS:MPI::MPI_C>
           $<$<BOOL:${RAPIDSMPF_HAVE_CUPTI}>:CUDA::cupti>

cpp/benchmarks/CMakeLists.txt

@@ -122,6 +122,32 @@ install(
   EXCLUDE_FROM_ALL
 )
 
+add_executable(bench_pinned_pool_fragmentation "bench_pinned_pool_fragmentation.cpp")
+set_target_properties(
+  bench_pinned_pool_fragmentation
+  PROPERTIES RUNTIME_OUTPUT_DIRECTORY "$<BUILD_INTERFACE:${RAPIDSMPF_BINARY_DIR}/benchmarks>"
+             CXX_STANDARD 20
+             CXX_STANDARD_REQUIRED ON
+             CXX_EXTENSIONS ON
+             CUDA_STANDARD 20
+             CUDA_STANDARD_REQUIRED ON
+)
+target_compile_options(
+  bench_pinned_pool_fragmentation PRIVATE "$<$<COMPILE_LANGUAGE:CXX>:${RAPIDSMPF_CXX_FLAGS}>"
+                                          "$<$<COMPILE_LANGUAGE:CUDA>:${RAPIDSMPF_CUDA_FLAGS}>"
+)
+target_link_libraries(
+  bench_pinned_pool_fragmentation
+  PRIVATE rapidsmpf::rapidsmpf rmm::rmm benchmark::benchmark benchmark::benchmark_main
+          $<TARGET_NAME_IF_EXISTS:conda_env> maybe_asan bench_utils
+)
+install(
+  TARGETS bench_pinned_pool_fragmentation
+  COMPONENT benchmarking
+  DESTINATION bin/benchmarks/librapidsmpf
+  EXCLUDE_FROM_ALL
+)
+
 add_executable(bench_pack "bench_pack.cpp")
 set_target_properties(
   bench_pack
@@ -148,6 +174,32 @@ install(
   EXCLUDE_FROM_ALL
 )
 
+add_executable(bench_driver_pool_fragmentation "bench_driver_pool_fragmentation.cpp")
+set_target_properties(
+  bench_driver_pool_fragmentation
+  PROPERTIES RUNTIME_OUTPUT_DIRECTORY "$<BUILD_INTERFACE:${RAPIDSMPF_BINARY_DIR}/benchmarks>"
+             CXX_STANDARD 20
+             CXX_STANDARD_REQUIRED ON
+             CXX_EXTENSIONS ON
+             CUDA_STANDARD 20
+             CUDA_STANDARD_REQUIRED ON
+)
+target_compile_options(
+  bench_driver_pool_fragmentation PRIVATE "$<$<COMPILE_LANGUAGE:CXX>:${RAPIDSMPF_CXX_FLAGS}>"
+                                          "$<$<COMPILE_LANGUAGE:CUDA>:${RAPIDSMPF_CUDA_FLAGS}>"
+)
+target_link_libraries(
+  bench_driver_pool_fragmentation
+  PRIVATE benchmark::benchmark benchmark::benchmark_main CUDA::cudart CCCL::CCCL
+          $<TARGET_NAME_IF_EXISTS:conda_env>
+)
+install(
+  TARGETS bench_driver_pool_fragmentation
+  COMPONENT benchmarking
+  DESTINATION bin/benchmarks/librapidsmpf
+  EXCLUDE_FROM_ALL
+)
+
 if(RAPIDSMPF_HAVE_STREAMING)
   add_subdirectory(streaming)
 endif()