null_mask.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2019-2025, NVIDIA CORPORATION.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #pragma once
17 
18 #include <cudf/types.hpp>
20 #include <cudf/utilities/export.hpp>
22 #include <cudf/utilities/span.hpp>
23 
24 #include <rmm/device_buffer.hpp>
25 
26 #include <vector>
27 
28 namespace CUDF_EXPORT cudf {
29 
48 
61 std::size_t bitmask_allocation_size_bytes(size_type number_of_bits,
62  std::size_t padding_boundary = 64);
63 
78 
91  size_type size,
92  mask_state state,
95 
110  size_type begin_bit,
111  size_type end_bit,
112  bool valid,
114 
133 [[deprecated]] void set_null_masks(cudf::host_span<bitmask_type*> bitmasks,
138 
159 
180 
199  bitmask_type const* mask,
200  size_type begin_bit,
201  size_type end_bit,
204 
218  column_view const& view,
221 
234 std::pair<rmm::device_buffer, size_type> bitmask_and(
235  table_view const& view,
238 
255 std::pair<std::vector<std::unique_ptr<rmm::device_buffer>>, std::vector<size_type>>
257  host_span<size_type const> segment_offsets,
260 
273 std::pair<rmm::device_buffer, size_type> bitmask_or(
274  table_view const& view,
277 
295  size_type start,
296  size_type stop,
297  rmm::cuda_stream_view stream = cudf::get_default_stream()); // end of group
299 } // namespace CUDF_EXPORT cudf
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
A set of cudf::column_view's of the same size.
Definition: table_view.hpp:200
std::pair< std::vector< std::unique_ptr< rmm::device_buffer > >, std::vector< size_type > > segmented_bitmask_and(host_span< column_view const > colviews, host_span< size_type const > segment_offsets, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Performs segmented bitwise AND operations on the null masks of the input columns based on defined seg...
void set_null_masks_safe(cudf::host_span< bitmask_type * > bitmasks, cudf::host_span< size_type const > begin_bits, cudf::host_span< size_type const > end_bits, cudf::host_span< bool const > valids, rmm::cuda_stream_view stream=cudf::get_default_stream())
Sets a vector of non-overlapping pre-allocated bitmask buffers to given states in the corresponding r...
cudf::size_type null_count(bitmask_type const *bitmask, size_type start, size_type stop, rmm::cuda_stream_view stream=cudf::get_default_stream())
Given a validity bitmask, counts the number of null elements (unset bits) in the range [start,...
std::size_t bitmask_allocation_size_bytes(size_type number_of_bits, std::size_t padding_boundary=64)
Computes the required bytes necessary to represent the specified number of bits with a given padding ...
void set_null_masks(cudf::host_span< bitmask_type * > bitmasks, cudf::host_span< size_type const > begin_bits, cudf::host_span< size_type const > end_bits, cudf::host_span< bool const > valids, rmm::cuda_stream_view stream=cudf::get_default_stream())
Sets a vector of non-overlapping pre-allocated bitmask buffers to given states in the corresponding n...
size_type state_null_count(mask_state state, size_type size)
Returns the null count for a null mask of the specified state representing size elements.
rmm::device_buffer copy_bitmask(column_view const &view, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Copies view's bitmask from the bits [view.offset(), view.offset() + view.size()) into a device_buffer
rmm::device_buffer create_null_mask(size_type size, mask_state state, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Creates a device_buffer for use as a null value indicator bitmask of a column
std::pair< rmm::device_buffer, size_type > bitmask_or(table_view const &view, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Performs bitwise OR of the bitmasks of columns of a table. Returns a pair of resulting mask and count...
void set_null_mask(bitmask_type *bitmask, size_type begin_bit, size_type end_bit, bool valid, rmm::cuda_stream_view stream=cudf::get_default_stream())
Sets a pre-allocated bitmask buffer to a given state in the range [begin_bit, end_bit)
std::pair< rmm::device_buffer, size_type > bitmask_and(table_view const &view, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Performs bitwise AND of the bitmasks of columns of a table. Returns a pair of resulting mask and coun...
void set_null_masks_unsafe(cudf::host_span< bitmask_type * > bitmasks, cudf::host_span< size_type const > begin_bits, cudf::host_span< size_type const > end_bits, cudf::host_span< bool const > valids, rmm::cuda_stream_view stream=cudf::get_default_stream())
Sets a vector of non-overlapping pre-allocated bitmask buffers to given states in the corresponding n...
size_type num_bitmask_words(size_type number_of_bits)
Returns the number of bitmask_type words required to represent the specified number of bits.
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
cuda::mr::async_resource_ref< cuda::mr::device_accessible > device_async_resource_ref
int32_t size_type
Row index type for columns and tables.
Definition: types.hpp:95
uint32_t bitmask_type
Bitmask type stored as 32-bit unsigned integer.
Definition: types.hpp:96
mask_state
Controls the allocation/initialization of a null mask.
Definition: types.hpp:181
cuDF interfaces
Definition: host_udf.hpp:37
APIs for spans.
C++20 std::span with reduced feature set.
Definition: span.hpp:194
Type declarations for libcudf.