19 lines
492 B
C++
19 lines
492 B
C++
|
|
#include <pybind11/pybind11.h>
|
|
|
|
#include "SpikeGenerationGPU.h"
|
|
|
|
namespace py = pybind11;
|
|
|
|
PYBIND11_MODULE(PySpikeGenerationGPU, m)
|
|
{
|
|
m.doc() = "SpikeGenerationGPU Module";
|
|
py::class_<SpikeGenerationGPU>(m, "SpikeGenerationGPU")
|
|
.def(py::init<>())
|
|
.def("gpu_occupancy_export",
|
|
&SpikeGenerationGPU::gpu_occupancy_export)
|
|
.def("gpu_occupancy_import",
|
|
&SpikeGenerationGPU::gpu_occupancy_import)
|
|
.def("spike_generation",
|
|
&SpikeGenerationGPU::entrypoint);
|
|
}
|