23 #pragma nv_diag_suppress 611
24 #pragma nv_diag_suppress 2810
26 #include <rmm/resource_ref.hpp>
28 #include <arrow/api.h>
30 #pragma nv_diag_default 611
31 #pragma nv_diag_default 2810
35 #include <cudf/detail/transform.hpp>
41 #include <rmm/mr/device/per_device_resource.hpp>
43 struct DLManagedTensor;
45 struct ArrowDeviceArray;
74 DLManagedTensor
const* managed_tensor,
75 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
97 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
147 std::vector<column_metadata>
const& metadata = {},
149 arrow::MemoryPool* ar_mr = arrow::default_memory_pool());
171 arrow::MemoryPool* ar_mr = arrow::default_memory_pool());
230 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
259 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
291 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
323 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
334 arrow::Table
const& input,
336 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
347 arrow::Scalar
const& input,
349 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
365 template <
typename ViewType>
389 std::unique_ptr<cudf::table_view, custom_view_deleter<cudf::table_view>>;
427 ArrowSchema
const* schema,
428 ArrowDeviceArray
const* input,
430 rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
437 std::unique_ptr<cudf::column_view, custom_view_deleter<cudf::column_view>>;
470 ArrowSchema
const* schema,
471 ArrowDeviceArray
const* input,
473 rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
A container of nullable device data as a column of elements.
An owning class to represent a singular value.
A set of cudf::column_view's of the same size.
A set of cudf::column's of the same size.
Class definition for cudf::column.
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
std::unique_ptr< cudf::table_view, custom_view_deleter< cudf::table_view > > unique_table_view_t
typedef for a unique_ptr to a cudf::table_view with custom deleter
std::vector< std::unique_ptr< cudf::column > > owned_columns_t
typedef for a vector of owning columns, used for conversion from ArrowDeviceArray
unique_table_view_t from_arrow_device(ArrowSchema const *schema, ArrowDeviceArray const *input, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Create cudf::table_view from given ArrowDeviceArray and ArrowSchema
unique_schema_t to_arrow_schema(cudf::table_view const &input, cudf::host_span< column_metadata const > metadata)
Create ArrowSchema from cudf table and metadata.
std::unique_ptr< cudf::column_view, custom_view_deleter< cudf::column_view > > unique_column_view_t
typedef for a unique_ptr to a cudf::column_view with custom deleter
std::unique_ptr< table > from_arrow(arrow::Table const &input, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=rmm::mr::get_current_device_resource())
Create cudf::table from given arrow Table input.
std::unique_ptr< ArrowSchema, void(*)(ArrowSchema *)> unique_schema_t
typedef for a unique_ptr to an ArrowSchema with custom deleter
unique_column_view_t from_arrow_device_column(ArrowSchema const *schema, ArrowDeviceArray const *input, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Create cudf::column_view from given ArrowDeviceArray and ArrowSchema
std::unique_ptr< ArrowDeviceArray, void(*)(ArrowDeviceArray *)> unique_device_array_t
typedef for a unique_ptr to an ArrowDeviceArray with a custom deleter
std::shared_ptr< arrow::Table > to_arrow(table_view input, std::vector< column_metadata > const &metadata={}, rmm::cuda_stream_view stream=cudf::get_default_stream(), arrow::MemoryPool *ar_mr=arrow::default_memory_pool())
Create arrow::Table from cudf table input
unique_device_array_t to_arrow_device(cudf::table &&table, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=rmm::mr::get_current_device_resource())
Create ArrowDeviceArray from cudf table and metadata.
DLManagedTensor * to_dlpack(table_view const &input, rmm::device_async_resource_ref mr=rmm::mr::get_current_device_resource())
Convert a cudf table into a DLPack DLTensor.
std::unique_ptr< table > from_dlpack(DLManagedTensor const *managed_tensor, rmm::device_async_resource_ref mr=rmm::mr::get_current_device_resource())
Convert a DLPack DLTensor into a cudf table.
functor for a custom deleter to a unique_ptr of table_view
void operator()(ViewType *ptr) const
operator to delete the unique_ptr
owned_columns_t owned_mem_
Owned columns that must be deleted.
custom_view_deleter(owned_columns_t &&owned)
Construct a new custom view deleter object.
C++20 std::span with reduced feature set.
Class definition for cudf::table.
Class definitions for (mutable)_table_view
Type declarations for libcudf.