hipdf.from_dlpack

Contents

hipdf.from_dlpack#

21 min read time

Applies to Linux

hipdf.from_dlpack(pycapsule_obj)#

Converts from a DLPack tensor to a cuDF object.

DLPack is an open-source memory tensor structure: dmlc/dlpack.

This function takes a PyCapsule object which contains a pointer to a DLPack tensor as input, and returns a cuDF object. This function deep copies the data in the DLPack tensor into a cuDF object.

Parameters#

pycapsule_objPyCapsule

Input DLPack tensor pointer which is encapsulated in a PyCapsule object.

Returns#

A cuDF DataFrame or Series depending on if the input DLPack tensor is 1D or 2D.

Notes#

cuDF from_dlpack() assumes column-major (Fortran order) input. If the input tensor is row-major, transpose it before passing it to this function.

Contents