pytorch-sbs/network/CPP_Cuda_new_preview/PyHDynamicCNNManyIP.cpp

18 lines
517 B
C++
Raw Permalink Normal View History

2023-01-13 21:33:57 +01:00
#include <pybind11/pybind11.h>
#include "HDynamicCNNManyIP.h"
namespace py = pybind11;
PYBIND11_MODULE(PyHDynamicCNNManyIP, m)
{
m.doc() = "HDynamicCNNManyIP Module";
py::class_<HDynamicCNNManyIP>(m, "HDynamicCNNManyIP")
.def(py::init<>())
.def("gpu_occupancy_export",
&HDynamicCNNManyIP::gpu_occupancy_export)
.def("gpu_occupancy_import",
&HDynamicCNNManyIP::gpu_occupancy_import)
.def("update",
&HDynamicCNNManyIP::update_entrypoint);
}