pytorch-sbs/network/spike_generation_cpu_cpp/PySpikeGenerationCPU.cpp
2023-02-04 14:26:14 +01:00

19 lines
492 B
C++

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