Skip to content

Add cudf table packing utilities with memory type support#843

Draft
nirandaperera wants to merge 5 commits intorapidsai:mainfrom
nirandaperera:better_cudf_pack_usage2
Draft

Add cudf table packing utilities with memory type support#843
nirandaperera wants to merge 5 commits intorapidsai:mainfrom
nirandaperera:better_cudf_pack_usage2

Conversation

@nirandaperera
Copy link
Contributor

@nirandaperera nirandaperera commented Feb 3, 2026

Introduces pack.hpp and pack.cpp with utilities for packing cudf tables into contiguous buffers with explicit memory type control. This enables packing directly to device, pinned host, or host memory using the rapidsmpf memory reservation system.

Depends on #840

Changes

New files:

  • cpp/include/rapidsmpf/integrations/cudf/pack.hpp
  • cpp/src/integrations/cudf/pack.cpp
  • cpp/tests/test_cudf_pack.cpp

API:

Function Description
detail::pack<MemoryType>() Templated pack with compile-time memory type
pack() Runtime dispatch based on reservation's memory type
chunked_pack() Uses cudf::chunked_pack with a bounce buffer for memory-efficient packing to host

Refactored:

  • TableChunk::copy() now uses detail::pack<> for host memory destinations

@nirandaperera nirandaperera requested review from a team as code owners February 3, 2026 23:41
@nirandaperera nirandaperera marked this pull request as draft February 3, 2026 23:41
@copy-pr-bot
Copy link

copy-pr-bot bot commented Feb 3, 2026

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

Signed-off-by: niranda perera <niranda.perera@gmail.com>
Signed-off-by: niranda perera <niranda.perera@gmail.com>
@nirandaperera nirandaperera force-pushed the better_cudf_pack_usage2 branch from 7dc13db to bdf915a Compare February 5, 2026 18:52
@nirandaperera nirandaperera added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels Feb 5, 2026
Signed-off-by: niranda perera <niranda.perera@gmail.com>
Signed-off-by: niranda perera <niranda.perera@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant