Program Listing for File cupy_util.hpp

Return to documentation for file (morpheus/_lib/include/morpheus/utilities/cupy_util.hpp)

Copy
Copied!
            

#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

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.