cudf::mark_join Class Reference

Mark-based hash join for semi/anti join with left table reuse. More...

#include <mark_join.hpp>

Public Member Functions
	mark_join (mark_join const &)=delete
	mark_join (mark_join &&)=delete
mark_join &	operator= (mark_join const &)=delete
mark_join &	operator= (mark_join &&)=delete
	mark_join (cudf::table_view const &build, cudf::null_equality compare_nulls=null_equality::EQUAL, rmm::cuda_stream_view stream=cudf::get_default_stream())
	Constructs a mark join object by building a hash table from the build table. More...
	mark_join (cudf::table_view const &build, cudf::null_equality compare_nulls, double load_factor, rmm::cuda_stream_view stream=cudf::get_default_stream())
	Constructs a mark join object with a specified load factor. More...
std::unique_ptr< rmm::device_uvector< size_type > >	semi_join (cudf::table_view const &probe, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref()) const
	Returns build row indices that have at least one match in the probe table. More...
std::unique_ptr< rmm::device_uvector< size_type > >	anti_join (cudf::table_view const &probe, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref()) const
	Returns build row indices that have no match in the probe table. More...

Detailed Description

Mark-based hash join for semi/anti join with left table reuse.

Builds a hash table from the build (left) table using a multiset that allows duplicate keys. The probe kernel atomically marks matching build entries via CAS on the hash MSB, then a retrieve kernel collects marked (semi) or unmarked (anti) entries.

This class enables building the hash table once and probing multiple times with different right (probe) tables, amortizing the build cost.

Note

This class is designed for the case where the left table is reused across multiple semi/anti join operations. It should only be used when:

• The left (build) table is smaller than the right (probe) table. Building a hash table from the larger table is memory-inefficient and leads to poor probe performance due to longer collision chains.
• The left table is reasonably small (e.g. ≤1M rows). The mark-based probe walks the hash table linearly per probe row, so performance degrades with large hash tables. For large build tables, consider alternative join strategies.
• The left table is probed multiple times. If only a single join is needed, there is no benefit to reuse and standard join APIs may be more efficient.

For the common case where the right (filter) table is reused, use cudf::filtered_join instead, which builds a distinct set from the right table.

Note

All NaNs are considered as equal

Definition at line 59 of file mark_join.hpp.

Constructor & Destructor Documentation

mark_join() [1/2]

cudf::mark_join::mark_join	(	cudf::table_view const &	build,
		cudf::null_equality	compare_nulls = null_equality::EQUAL,
		rmm::cuda_stream_view	stream = cudf::get_default_stream()
	)

Constructs a mark join object by building a hash table from the build table.

Parameters

build	The build table (typically the left table)
compare_nulls	Controls whether null join-key values should match or not
stream	CUDA stream used for device memory operations and kernel launches

mark_join() [2/2]

cudf::mark_join::mark_join	(	cudf::table_view const &	build,
		cudf::null_equality	compare_nulls,
		double	load_factor,
		rmm::cuda_stream_view	stream = cudf::get_default_stream()
	)

Constructs a mark join object with a specified load factor.

Parameters

build	The build table (typically the left table)
compare_nulls	Controls whether null join-key values should match or not
load_factor	Hash table load factor in range (0,1]
stream	CUDA stream used for device memory operations and kernel launches

Member Function Documentation

anti_join()

std::unique_ptr<rmm::device_uvector<size_type> > cudf::mark_join::anti_join	(	cudf::table_view const &	probe,
		rmm::cuda_stream_view	stream = cudf::get_default_stream(),
		rmm::device_async_resource_ref	mr = cudf::get_current_device_resource_ref()
	)		const

Returns build row indices that have no match in the probe table.

Parameters

probe	The probe table
stream	CUDA stream used for device memory operations and kernel launches
mr	Device memory resource used to allocate the returned device memory

Returns

Device vector of build row indices

semi_join()

std::unique_ptr<rmm::device_uvector<size_type> > cudf::mark_join::semi_join	(	cudf::table_view const &	probe,
		rmm::cuda_stream_view	stream = cudf::get_default_stream(),
		rmm::device_async_resource_ref	mr = cudf::get_current_device_resource_ref()
	)		const

Returns build row indices that have at least one match in the probe table.

Parameters

probe	The probe table
stream	CUDA stream used for device memory operations and kernel launches
mr	Device memory resource used to allocate the returned device memory

Returns

Device vector of build row indices

---

The documentation for this class was generated from the following file:

• mark_join.hpp