Copy, Serialization, and Utility APIs

raft::copy

Source header: raft/core/copy.hpp

Asynchronously copies elements between compatible memory locations.

1
template <typename OutputIterator, typename InputIterator, typename SizeType>
2
void copy(OutputIterator dst, InputIterator src, SizeType n,
3
          rmm::cuda_stream_view stream);

Parameters

Returns

void

raft::copy_matrix

Source header: raft/core/copy.hpp

Copies a dense matrix between compatible matrix views.

1
template <typename OutputView, typename InputView>
2
void copy_matrix(OutputView dst, InputView src, rmm::cuda_stream_view stream);

Parameters

Returns

void

raft::update_device

Source header: raft/core/copy.hpp

Convenience helper for copying host data to device memory.

1
template <typename DevicePointer, typename HostPointer, typename SizeType>
2
void update_device(DevicePointer dst, HostPointer src, SizeType n,
3
                   rmm::cuda_stream_view stream);

Parameters

Returns

void

raft::update_host

Source header: raft/core/copy.hpp

Convenience helper for copying device data to host memory.

1
template <typename HostPointer, typename DevicePointer, typename SizeType>
2
void update_host(HostPointer dst, DevicePointer src, SizeType n,
3
                 rmm::cuda_stream_view stream);

Parameters

Returns

void

raft::serialize_mdspan

Source header: raft/core/serialize.hpp

Serializes array contents from an mdspan-like view.

1
template <typename View>
2
void serialize_mdspan(std::ostream& os, View view);

Parameters

Returns

void

raft::deserialize_mdspan

Source header: raft/core/serialize.hpp

Deserializes array contents into an mdspan-like view.

1
template <typename View>
2
void deserialize_mdspan(std::istream& is, View view);

Parameters

Returns

void

raft::ceildiv

Source header: raft/util/integer_utils.hpp

Computes integer division rounded up.

1
template <typename T>
2
T ceildiv(T dividend, T divisor);

Parameters

Returns

T

raft::round_up_safe

Source header: raft/util/integer_utils.hpp

Rounds an integer value up to a requested multiple.

1
template <typename T>
2
T round_up_safe(T value, T multiple);

Parameters

Returns

T