18 lines
486 B
C++
18 lines
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);
|
||
|
|
}
|