mirror of
https://github.com/davrot/pytorch-sbs.git
synced 2025-07-06 20:00:02 +02:00
19 lines
492 B
C++
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);
|
|
}
|