Add files via upload

network/count_spikes_cpu_cpp/CountSpikesCPU.cpp

@@ -0,0 +1,119 @@
+#include "CountSpikesCPU.h"
+
+#include <omp.h>
+#include <stdio.h>
+#include <string.h>
+
+#include <algorithm>
+#include <cassert>
+#include <iostream>
+
+
+CountSpikesCPU::CountSpikesCPU()
+{
+
+};
+
+CountSpikesCPU::~CountSpikesCPU()
+{
+
+};
+
+void CountSpikesCPU::process(
+    int64_t input_pointer_addr,
+    int64_t input_dim_0,
+    int64_t input_dim_1,
+    int64_t input_dim_2,
+    int64_t input_dim_3,
+    int64_t output_pointer_addr,
+    int64_t output_dim_1,
+    int64_t number_of_cpu_processes)
+{
+    int64_t* input_pointer = (int64_t*)input_pointer_addr;
+    int64_t* output_pointer = (int64_t*)output_pointer_addr;
+
+    // Input
+    assert((input_pointer != nullptr));
+    assert((input_dim_0 > 0));
+    assert((input_dim_1 > 0));
+    assert((input_dim_2 > 0));
+    assert((input_dim_3 > 0));
+
+    // Output
+    assert((output_pointer != nullptr));
+    assert((output_dim_1 > 0));
+
+    // Input
+    size_t input_dim_c0 = input_dim_1 * input_dim_2 * input_dim_3;
+    size_t dim_c1 = input_dim_2 * input_dim_3;
+    size_t dim_c2 = input_dim_3;
+
+    // Output
+    size_t output_dim_c0 = output_dim_1 * input_dim_2 * input_dim_3;
+
+    assert((number_of_cpu_processes > 0));
+
+    omp_set_num_threads(number_of_cpu_processes);
+    // DEBUG:
+    // omp_set_num_threads(1);
+
+#pragma omp parallel for
+    for (size_t pattern_id = 0; pattern_id < input_dim_0; pattern_id++)
+    {
+        process_pattern(
+            input_pointer + input_dim_c0 * pattern_id,
+            input_dim_1,
+            input_dim_2,
+            input_dim_3,
+            output_pointer + output_dim_c0 * pattern_id,
+            output_dim_1,
+            dim_c1,
+            dim_c2,
+            pattern_id);
+    }
+
+
+    return;
+};
+
+void CountSpikesCPU::process_pattern(
+    int64_t* input_pointer,
+    size_t input_dim_1,
+    size_t input_dim_2,
+    size_t input_dim_3,
+    int64_t* output_pointer,
+    size_t output_dim_1,
+    size_t dim_c1,
+    size_t dim_c2,
+    size_t pattern_id)
+{
+    size_t position = 0;
+    int64_t* input_pointer_x;
+    int64_t* input_pointer_xy;
+    int64_t* output_pointer_x;
+    int64_t* output_pointer_xy;
+
+    for (size_t x_id = 0; x_id < input_dim_2; x_id++)
+    {
+        input_pointer_x = input_pointer + x_id * dim_c2;
+        output_pointer_x = output_pointer +  x_id * dim_c2;
+
+        for (size_t y_id = 0; y_id < input_dim_3; y_id++)
+        {
+            input_pointer_xy = input_pointer_x + y_id;
+            output_pointer_xy = output_pointer_x + y_id;
+
+            for (size_t sp_id = 0; sp_id < input_dim_1; sp_id++)
+            {
+                position = input_pointer_xy[sp_id * dim_c1];
+                if ((position >= 0) && (position < output_dim_1))
+                {
+
+                    output_pointer_xy[position * dim_c1] ++;
+                }
+            }
+        }
+    }
+
+    return;
+};

network/count_spikes_cpu_cpp/CountSpikesCPU.h

@@ -0,0 +1,39 @@
+#ifndef COUNTSPIKESCPU
+#define COUNTSPIKESCPU
+
+#include <unistd.h>
+
+#include <cctype>
+#include <iostream>
+
+class CountSpikesCPU
+{
+    public:
+    CountSpikesCPU();
+    ~CountSpikesCPU();
+
+    void process(
+        int64_t input_pointer_addr,
+        int64_t input_dim_0,
+        int64_t input_dim_1,
+        int64_t input_dim_2,
+        int64_t input_dim_3,
+        int64_t output_pointer_addr,
+        int64_t output_dim_1,
+        int64_t number_of_cpu_processes);
+
+    private:
+    void process_pattern(
+        int64_t* input_pointer,
+        size_t input_dim_1,
+        size_t input_dim_2,
+        size_t input_dim_3,
+        int64_t* output_pointer,
+        size_t output_dim_1,
+        size_t dim_c1,
+        size_t dim_c2,
+        size_t pattern_id);
+
+};
+
+#endif /* COUNTSPIKESCPU */

network/count_spikes_cpu_cpp/Makefile

@@ -0,0 +1,33 @@
+include ../.env
+export
+
+name = CountSpikes
+type = CPU
+
+PYPOSTFIX := $(shell $(PYBIN)python3-config --extension-suffix)
+PYBIND11INCLUDE := $(shell $(PYBIN)python3 -m pybind11 --includes)
+PARAMETERS_O = $(PARAMETERS_O_CPU) $(PYBIND11INCLUDE) 
+PARAMETERS_Linker = $(PARAMETERS_Linker_CPU)
+
+so_file = Py$(name)$(type)$(PYPOSTFIX)
+pyi_file = Py$(name)$(type).pyi
+all: ../$(so_file)
+
+$(O_DIRS)$(name)$(type).o: $(name)$(type).h $(name)$(type).cpp
+	mkdir -p $(O_DIRS) 
+	$(CC) $(PARAMETERS_O) -c $(name)$(type).cpp -o $(O_DIRS)$(name)$(type).o
+
+$(O_DIRS)Py$(name)$(type).o: $(name)$(type).h Py$(name)$(type).cpp 
+	mkdir -p $(O_DIRS)
+	$(CC) $(PARAMETERS_O) -c Py$(name)$(type).cpp -o $(O_DIRS)Py$(name)$(type).o
+
+../$(so_file): $(O_DIRS)$(name)$(type).o $(O_DIRS)Py$(name)$(type).o
+	$(CC) $(PARAMETERS_Linker) -o ../$(so_file) $(O_DIRS)$(name)$(type).o $(O_DIRS)Py$(name)$(type).o
+
+
+#######################
+clean:
+	rm -rf $(O_DIRS)
+	rm -f ../$(so_file)
+	rm -f ../$(pyi_file)
+

network/count_spikes_cpu_cpp/PyCountSpikesCPU.cpp

@@ -0,0 +1,15 @@
+
+#include <pybind11/pybind11.h>
+
+#include "CountSpikesCPU.h"
+
+namespace py = pybind11;
+
+PYBIND11_MODULE(PyCountSpikesCPU, m)
+{
+  m.doc() = "CountSpikesCPU Module";
+  py::class_<CountSpikesCPU>(m, "CountSpikesCPU")
+    .def(py::init<>())
+    .def("process",
+      &CountSpikesCPU::process);
+}