Files | Functions

Files

file  interop.hpp
 

Functions

std::unique_ptr< tablecudf::from_dlpack (DLManagedTensor const *managed_tensor, rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
 Convert a DLPack DLTensor into a cudf table. More...
 
DLManagedTensor * cudf::to_dlpack (table_view const &input, rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
 Convert a cudf table into a DLPack DLTensor. More...
 

Detailed Description

Function Documentation

◆ from_dlpack()

std::unique_ptr<table> cudf::from_dlpack ( DLManagedTensor const *  managed_tensor,
rmm::mr::device_memory_resource mr = rmm::mr::get_current_device_resource() 
)

Convert a DLPack DLTensor into a cudf table.

The device_type of the DLTensor must be kDLGPU, kDLCPU, or kDLCPUPinned, and device_id must match the current device. The ndim must be set to 1 or 2. The dtype must have 1 lane and the bitsize must match a supported cudf::data_type.

Note
The managed tensor is not deleted by this function.
Exceptions
cudf::logic_errorif the any of the DLTensor fields are unsupported
Parameters
managed_tensora 1D or 2D column-major (Fortran order) tensor
mrDevice memory resource used to allocate the returned table's device memory.
Returns
Table with a copy of the tensor data

◆ to_dlpack()

DLManagedTensor* cudf::to_dlpack ( table_view const &  input,
rmm::mr::device_memory_resource mr = rmm::mr::get_current_device_resource() 
)

Convert a cudf table into a DLPack DLTensor.

All columns must have the same data type and this type must be numeric. The columns may be nullable, but the null count must be zero. If the input table is empty or has zero rows, the result will be nullptr.

Note
The deleter method of the returned DLManagedTensor must be used to free the memory allocated for the tensor.
Exceptions
cudf::logic_errorif the data types are not equal or not numeric, or if any of columns have non-zero null count
Parameters
inputTable to convert to DLPack
mrDevice memory resource used to allocate the returned DLPack tensor's device memory.
Returns
1D or 2D DLPack tensor with a copy of the table data, or nullptr