15 #include <cudf/utilities/export.hpp>
27 namespace CUDF_EXPORT
cudf {
250 std::vector<std::reference_wrapper<scalar const>>
const& source,
495 std::span<size_type const> indices,
502 std::initializer_list<size_type> indices,
539 std::span<size_type const> indices,
546 std::initializer_list<size_type> indices,
583 std::span<size_type const> splits,
590 std::initializer_list<size_type> splits,
629 std::span<size_type const> splits,
636 std::initializer_list<size_type> splits,
819 std::vector<std::reference_wrapper<scalar const>>
const& input,
886 int64_t
const seed = 0,
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
Non-owning references to the memory resources used by a cuDF operation.
A non-owning, mutable view of device data as a column of elements, some of which may be null as indic...
An owning class to represent a singular value.
A set of cudf::column_view's of the same size.
column view class definitions
mask_allocation_policy
Indicates when to allocate a mask, based on an existing mask.
std::unique_ptr< column > purge_nonempty_nulls(column_view const &input, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Copy input into output while purging any non-empty null rows in the column or its descendants.
std::unique_ptr< column > copy_range(column_view const &source, column_view const &target, size_type source_begin, size_type source_end, size_type target_begin, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Copies a range of elements out-of-place from one column to another.
bool may_have_nonempty_nulls(column_view const &input)
Approximates if a column or its descendants may have non-empty null elements.
void copy_range_in_place(column_view const &source, mutable_column_view &target, size_type source_begin, size_type source_end, size_type target_begin, cuda::stream_ref stream=cudf::get_default_stream())
Copies a range of elements in-place from one column to another.
std::unique_ptr< column > allocate_like(column_view const &input, size_type size, mask_allocation_policy mask_alloc=mask_allocation_policy::RETAIN, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Creates an uninitialized new column of the specified size and same type as the input.
std::unique_ptr< table > sample(table_view const &input, size_type const n, sample_with_replacement replacement=sample_with_replacement::FALSE, int64_t const seed=0, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Gather n samples from given input randomly.
negative_index_policy
Policy to recognize negative indices as relative to the end of the source.
std::unique_ptr< column > reverse(column_view const &source_column, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Reverses the elements of a column.
std::unique_ptr< column > copy_if_else(scalar const &lhs, scalar const &rhs, column_view const &boolean_mask, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Returns a new column, where each element is selected from either lhs or rhs based on the value of the...
out_of_bounds_policy
Policy to account for possible out-of-bounds indices.
sample_with_replacement
Indicates whether a row can be sampled more than once.
std::unique_ptr< scalar > get_element(column_view const &input, size_type index, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Get the element at specified index from a column.
std::unique_ptr< table > empty_like(table_view const &input_table)
Creates a table of empty columns with the same types as the input_table
bool has_nonempty_nulls(column_view const &input, cuda::stream_ref stream=cudf::get_default_stream())
Checks if a column or its descendants have non-empty null rows.
@ RETAIN
Allocate a null mask if the input contains one.
@ NOT_ALLOWED
Negative indices are not allowed and result in undefined behavior.
@ ALLOWED
Negative indices are allowed and result in wrapping behavior.
@ DONT_CHECK
No bounds checking is performed, better performance.
@ NULLIFY
Output values corresponding to out-of-bounds indices are null.
@ FALSE
A row can be sampled only once.
@ TRUE
A row can be sampled more than once.
std::unique_ptr< table > gather(table_view const &source_table, column_view const &gather_map, out_of_bounds_policy bounds_policy, negative_index_policy neg_indices, cuda::stream_ref stream=cudf::get_default_stream(), cudf::memory_resources mr=cudf::get_current_device_resource_ref())
Gathers the specified rows of a set of columns according to a gather map.
std::unique_ptr< table > scatter(std::vector< std::reference_wrapper< scalar const >> const &source, column_view const &indices, table_view const &target, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Scatters a row of scalar values into a copy of the target table according to a scatter map.
std::unique_ptr< table > boolean_mask_scatter(std::vector< std::reference_wrapper< scalar const >> const &input, table_view const &target, column_view const &boolean_mask, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Scatters scalar values to rows of the output corresponding to true values in a boolean mask.
std::unique_ptr< column > shift(column_view const &input, size_type offset, scalar const &fill_value, cuda::stream_ref stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Creates a new column by shifting all values by an offset.
std::vector< table_view > slice(table_view const &input, std::initializer_list< size_type > indices, cuda::stream_ref stream=cudf::get_default_stream())
Slices a table_view into a set of table_views according to a set of indices.
std::vector< table_view > split(table_view const &input, std::initializer_list< size_type > splits, cuda::stream_ref stream=cudf::get_default_stream())
Splits a table_view into a set of table_views according to a set of indices derived from expected spl...
cuda::stream_ref const get_default_stream()
Get the current default stream.
@ ALWAYS
Use dictionary regardless of impact on compression.
@ NEVER
Never use dictionary encoding.
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
cuda::mr::resource_ref< cuda::mr::device_accessible > device_async_resource_ref
int32_t size_type
Row index type for columns and tables.
Class definition for cudf::lists_column_view.
APIs for getting and setting the current device memory resource.
Class definitions for cudf::scalar.
Class definition for cudf::strings_column_view.
Class definition for cudf::structs_column_view.
Class definition for cudf::table.
Type declarations for libcudf.
