Program Listing for File 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