pytorch-sbs/network/CPP_Cuda/PySpikeGeneration2DManyIP.cpp
2023-01-29 00:59:05 +01:00

19 lines
549 B
C++

#include <pybind11/pybind11.h>
#include "SpikeGeneration2DManyIP.h"
namespace py = pybind11;
PYBIND11_MODULE(PySpikeGeneration2DManyIP, m)
{
m.doc() = "SpikeGeneration2DManyIP Module";
py::class_<SpikeGeneration2DManyIP>(m, "SpikeGeneration2DManyIP")
.def(py::init<>())
.def("gpu_occupancy_export",
&SpikeGeneration2DManyIP::gpu_occupancy_export)
.def("gpu_occupancy_import",
&SpikeGeneration2DManyIP::gpu_occupancy_import)
.def("spike_generation",
&SpikeGeneration2DManyIP::spike_generation_entrypoint);
}