Add partitioned probe support for hash joins#22108

Merged

rapids-bot[bot] merged 30 commits into

rapidsai:release/26.06from

PointKernel:chunked-hash-join-probe

May 20, 2026

Commits on May 19, 2026

Add partitioned probe support for hash joins
PointKernel
committed
Clean up rmm policy mr usage
PointKernel
committed
Use custom kernels + file split
PointKernel
committed
Remove redundant thrust::reduce in partitioned join retrieve
PointKernel
committed
Use DEFAULT_JOIN_BLOCK_SIZE and DEFAULT_JOIN_CG_SIZE directly
PointKernel
committed
Minor cleanup with modern CCCL
PointKernel
committed
Cleanups
PointKernel
committed
Improve retrieve kenrel perf with shared memory buffer
PointKernel
committed
Replace full_join_complement with full_join_finalize
PointKernel
committed
Use a common full join finalize logic
PointKernel
committed
Speed up full_join finalize with consume-in-place and fused compact
PointKernel
committed
Rename full_join_finalize -> finalize_partitioned_full_join; sort CMakeLists
PointKernel
committed
Address lamarrr review: explicit mode check in benchmark, delete copy ops
PointKernel
committed
Use cooperative_groups::invoke_one instead of thread_rank() == 0
PointKernel
committed
Rename count_each -> partitioned_count kernel and associated files
PointKernel
committed
Rename retrieve -> partitioned_retrieve kernel and associated files; add kernel doc
PointKernel
committed
Rename full_join_finalize.cpp -> partitioned_full_join_finalize.cpp
PointKernel
committed
Rename to finalize_partitioned_full_join.cpp to match API name
PointKernel
committed
Clean up stale comments: remove ported language, fix exclusive scan -> reduce
PointKernel
committed
Add doc comment to partitioned_count_kernel
PointKernel
committed
Formatting
PointKernel
committed
Address shrshi review nits and rename cnt -> match_count
PointKernel
committed
Replace raw int64_t with thread_index_type
PointKernel
committed
Use left/right naming
PointKernel
committed
Address CodeRabbit review feedback
PointKernel
committed
Fix probe_table_num_rows reference in match_context.cu
PointKernel
committed
Use right/left naming for table nouns in finalize_full_join and tests
PointKernel
committed

Commits on May 20, 2026

Simplify partitioned_count_kernel: drop tile.all short-circuit, fuse IsOuter writers
PointKernel
committed
Add tile size check + use cuda::std::distance
PointKernel
committed
Merge remote-tracking branch 'upstream/release/26.06' into chunked-hash-join-probe
PointKernel
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Add partitioned probe support for hash joins#22108

Add partitioned probe support for hash joins#22108
rapids-bot[bot] merged 30 commits into
rapidsai:release/26.06from
PointKernel:chunked-hash-join-probe

Commits on May 19, 2026

Add partitioned probe support for hash joins

Clean up rmm policy mr usage

Use custom kernels + file split

Remove redundant thrust::reduce in partitioned join retrieve

Use DEFAULT_JOIN_BLOCK_SIZE and DEFAULT_JOIN_CG_SIZE directly

Minor cleanup with modern CCCL

Cleanups

Improve retrieve kenrel perf with shared memory buffer

Replace full_join_complement with full_join_finalize

Use a common full join finalize logic

Speed up full_join finalize with consume-in-place and fused compact

Rename full_join_finalize -> finalize_partitioned_full_join; sort CMakeLists

Address lamarrr review: explicit mode check in benchmark, delete copy ops

Use cooperative_groups::invoke_one instead of thread_rank() == 0

Rename count_each -> partitioned_count kernel and associated files

Rename retrieve -> partitioned_retrieve kernel and associated files; add kernel doc

Rename full_join_finalize.cpp -> partitioned_full_join_finalize.cpp

Rename to finalize_partitioned_full_join.cpp to match API name

Clean up stale comments: remove ported language, fix exclusive scan -> reduce

Add doc comment to partitioned_count_kernel

Formatting

Address shrshi review nits and rename cnt -> match_count

Replace raw int64_t with thread_index_type

Use left/right naming

Address CodeRabbit review feedback

Fix probe_table_num_rows reference in match_context.cu

Use right/left naming for table nouns in finalize_full_join and tests

Commits on May 20, 2026

Simplify partitioned_count_kernel: drop tile.all short-circuit, fuse IsOuter writers

Add tile size check + use cuda::std::distance

Merge remote-tracking branch 'upstream/release/26.06' into chunked-hash-join-probe