↰ Return to documentation for file (morpheus/_lib/include/morpheus/utilities/cupy_util.hpp
)
#pragma once
#include "morpheus/objects/tensor_object.hpp"
#include <pybind11/pybind11.h>
#include <pybind11/pytypes.h>
namespace morpheus {
/****** Component public implementations *******************/
/****** CupyUtil****************************************/
struct CupyUtil
{
static pybind11::object cp_module; // handle to cupy module
static pybind11::module_ get_cp();
static pybind11::object tensor_to_cupy(const TensorObject &tensor);
static TensorObject cupy_to_tensor(pybind11::object cupy_array);
}; // end of group
} // namespace morpheus