transform.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2019-2025, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 
6 #pragma once
7 
8 #include <cudf/ast/expressions.hpp>
9 #include <cudf/types.hpp>
10 #include <cudf/utilities/export.hpp>
11 #include <cudf/utilities/memory_resource.hpp>
12 
13 #include <memory>
14 
15 namespace CUDF_EXPORT cudf {
16 
54 std::unique_ptr<column> transform(
55  std::vector<column_view> const& inputs,
56  std::string const& transform_udf,
57  data_type output_type,
58  bool is_ptx,
59  std::optional<void*> user_data = std::nullopt,
60  null_aware is_null_aware = null_aware::NO,
61  rmm::cuda_stream_view stream = cudf::get_default_stream(),
62  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
63 
76 std::pair<std::unique_ptr<rmm::device_buffer>, size_type> nans_to_nulls(
77  column_view const& input,
78  rmm::cuda_stream_view stream = cudf::get_default_stream(),
79  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
80 
95 std::unique_ptr<column> compute_column(
96  table_view const& table,
97  ast::expression const& expr,
98  rmm::cuda_stream_view stream = cudf::get_default_stream(),
99  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
100 
116 std::unique_ptr<column> compute_column_jit(
117  table_view const& table,
118  ast::expression const& expr,
119  rmm::cuda_stream_view stream = cudf::get_default_stream(),
120  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
121 
138 std::pair<std::unique_ptr<rmm::device_buffer>, cudf::size_type> bools_to_mask(
139  column_view const& input,
140  rmm::cuda_stream_view stream = cudf::get_default_stream(),
141  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
142 
168 std::pair<std::unique_ptr<cudf::table>, std::unique_ptr<cudf::column>> encode(
169  cudf::table_view const& input,
170  rmm::cuda_stream_view stream = cudf::get_default_stream(),
171  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
172 
201 std::pair<std::unique_ptr<column>, table_view> one_hot_encode(
202  column_view const& input,
203  column_view const& categories,
204  rmm::cuda_stream_view stream = cudf::get_default_stream(),
205  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
206 
229 std::unique_ptr<column> mask_to_bools(
230  bitmask_type const* bitmask,
231  size_type begin_bit,
232  size_type end_bit,
233  rmm::cuda_stream_view stream = cudf::get_default_stream(),
234  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
235 
262 std::unique_ptr<column> row_bit_count(
263  table_view const& t,
264  rmm::cuda_stream_view stream = cudf::get_default_stream(),
265  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
266 
286 std::unique_ptr<column> segmented_row_bit_count(
287  table_view const& t,
288  size_type segment_length,
289  rmm::cuda_stream_view stream = cudf::get_default_stream(),
290  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
291  // end of group
293 } // namespace CUDF_EXPORT cudf
cudf::column_view
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
Definition: column_view.hpp:316
cudf::data_type
Indicator for the logical data type of an element in a column.
Definition: types.hpp:238
cudf::table_view
A set of cudf::column_view's of the same size.
Definition: table_view.hpp:189
cudf::table
A set of cudf::column's of the same size.
Definition: table.hpp:29
rmm::cuda_stream_view
cudf::get_default_stream
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
cudf::get_current_device_resource_ref
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
Definition: memory_resource.hpp:40
device_async_resource_ref
detail::cccl_async_resource_ref< cuda::mr::resource_ref< cuda::mr::device_accessible > > device_async_resource_ref
cudf::encode
std::pair< std::unique_ptr< cudf::table >, std::unique_ptr< cudf::column > > encode(cudf::table_view const &input, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Encode the rows of the given table as integers.
cudf::row_bit_count
std::unique_ptr< column > row_bit_count(table_view const &t, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Returns an approximate cumulative size in bits of all columns in the table_view for each row.
cudf::transform
std::unique_ptr< column > transform(std::vector< column_view > const &inputs, std::string const &transform_udf, data_type output_type, bool is_ptx, std::optional< void * > user_data=std::nullopt, null_aware is_null_aware=null_aware::NO, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Creates a new column by applying a transform function against every element of the input columns.
cudf::bools_to_mask
std::pair< std::unique_ptr< rmm::device_buffer >, cudf::size_type > bools_to_mask(column_view const &input, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Creates a bitmask from a column of boolean elements.
cudf::one_hot_encode
std::pair< std::unique_ptr< column >, table_view > one_hot_encode(column_view const &input, column_view const &categories, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Encodes input by generating a new column for each value in categories indicating the presence of that...
cudf::mask_to_bools
std::unique_ptr< column > mask_to_bools(bitmask_type const *bitmask, size_type begin_bit, size_type end_bit, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Creates a boolean column from given bitmask.
cudf::nans_to_nulls
std::pair< std::unique_ptr< rmm::device_buffer >, size_type > nans_to_nulls(column_view const &input, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Creates a null_mask from input by converting NaN to null and preserving existing null values and also...
cudf::compute_column
std::unique_ptr< column > compute_column(table_view const &table, ast::expression const &expr, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Compute a new column by evaluating an expression tree on a table.
cudf::segmented_row_bit_count
std::unique_ptr< column > segmented_row_bit_count(table_view const &t, size_type segment_length, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Returns an approximate cumulative size in bits of all columns in the table_view for each segment of r...
cudf::compute_column_jit
std::unique_ptr< column > compute_column_jit(table_view const &table, ast::expression const &expr, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Compute a new column by evaluating an expression tree on a table using a JIT-compiled kernel.
cudf::size_type
int32_t size_type
Row index type for columns and tables.
Definition: types.hpp:84
cudf::bitmask_type
uint32_t bitmask_type
Bitmask type stored as 32-bit unsigned integer.
Definition: types.hpp:85
cudf::null_aware
null_aware
Indicates whether a function is null-aware or not.
Definition: types.hpp:227
memory_resource.hpp
cudf
cuDF interfaces
Definition: host_udf.hpp:26
cudf::ast::expression
A generic expression that can be evaluated to return a value.
Definition: expressions.hpp:61
types.hpp
Type declarations for libcudf.
Generated by doxygen 1.9.1