Dlpack Module

Rust module: cuvs::dlpack

Source: rust/cuvs/src/dlpack.rs

ManagedTensor

1
#[derive(Debug)]
2
pub struct ManagedTensor(ffi::DLManagedTensor); {
3
    /* private fields */
4
}

ManagedTensor is a wrapper around a dlpack DLManagedTensor object. This lets you pass matrices in device or host memory into cuvs.

Methods

as_ptr

1
pub fn as_ptr(&self) -> *mut ffi::DLManagedTensor

Source: rust/cuvs/src/dlpack.rs:21

to_device

1
pub fn to_device(&self, res: &Resources) -> Result<ManagedTensor>

Creates a new ManagedTensor on the current GPU device, and copies the data into it.

Source: rust/cuvs/src/dlpack.rs:27

to_host

1
pub fn to_host<
2
T: IntoDtype,
3
S: ndarray::RawData<Elem = T> + ndarray::RawDataMut,
4
D: ndarray::Dimension,
5
>(
6
&self,
7
res: &Resources,
8
arr: &mut ndarray::ArrayBase<S, D>,
9
) -> Result<()>

Copies data from device memory into host memory

Source: rust/cuvs/src/dlpack.rs:47

Source: rust/cuvs/src/dlpack.rs:14

IntoDtype

1
pub trait IntoDtype {
2
    /* required methods omitted */
3
}

Source: rust/cuvs/src/dlpack.rs:16