pylibraft.common.cai_wrapper

1 min read time

Applies to Linux

Classes

cai_wrapper

Simple wrapper around a CUDA array interface object to reduce

Functions

wrap_array(array)

Module Contents

class pylibraft.common.cai_wrapper.cai_wrapper(cai_arr)

Bases: pylibraft.common.ai_wrapper.ai_wrapper

Simple wrapper around a CUDA array interface object to reduce boilerplate for extracting common information from the underlying dictionary.

pylibraft.common.cai_wrapper.wrap_array(array)