pybind array_t benchmark added

Author: JohanMabille

benchmark/benchmark_pyarray2.py

@@ -0,0 +1,6 @@
+from xtensor_python_benchmark import pybind_sum_array
+import numpy as np
+
+u = np.ones(1000000, dtype=float)
+from timeit import timeit
+print (timeit ('pybind_sum_array(u)', setup='from __main__ import u, pybind_sum_array', number=1000))

benchmark/main.cpp

@@ -1,4 +1,5 @@
 #include "pybind11/pybind11.h"
+#include "pybind11/numpy.h"
 #include "xtensor/xtensor.hpp"
 #include "xtensor/xarray.hpp"
 #include "xtensor-python/pyarray.hpp"
@@ -19,5 +20,15 @@ PYBIND11_PLUGIN(xtensor_python_benchmark)
         }
     );
 
+    m.def("pybind_sum_array", [](py::array_t<double> const& x) {
+            double sum = 0;
+            size_t size = x.size();
+            const double* data = x.data(0);
+            for(size_t i = 0; i < size; ++i)
+                sum += data[i];
+            return sum;
+        }
+    );
+
     return m.ptr();
 }