pytorch-sbs/network/h_dynamic_cnn_gpu_cpp/PyHDynamicCNNGPU.cpp
2023-02-02 19:15:23 +01:00

18 lines
No EOL
486 B
C++

#include <pybind11/pybind11.h>
#include "HDynamicCNNGPU.h"
namespace py = pybind11;
PYBIND11_MODULE(PyHDynamicCNNGPU, m)
{
m.doc() = "HDynamicCNNGPU Module";
py::class_<HDynamicCNNGPU>(m, "HDynamicCNNGPU")
.def(py::init<>())
.def("gpu_occupancy_export",
&HDynamicCNNGPU::gpu_occupancy_export)
.def("gpu_occupancy_import",
&HDynamicCNNGPU::gpu_occupancy_import)
.def("update",
&HDynamicCNNGPU::entrypoint);
}