1 files changed, 93 insertions, 0 deletions
diff --git a/src/boost/libs/numeric/ublas/benchmarks/opencl/add.cpp b/src/boost/libs/numeric/ublas/benchmarks/opencl/add.cpp
new file mode 100644
index 000000000..b998cd2c0
--- /dev/null
+++ b/src/boost/libs/numeric/ublas/benchmarks/opencl/add.cpp
@@ -0,0 +1,93 @@
+//
+// Copyright (c) 2018 Stefan Seefeld
+//
+// Distributed under the Boost Software License, Version 1.0.
+// (See accompanying file LICENSE_1_0.txt or
+// copy at http://www.boost.org/LICENSE_1_0.txt)
+
+#define BOOST_UBLAS_ENABLE_OPENCL
+#include <boost/numeric/ublas/opencl.hpp>
+#include <boost/program_options.hpp>
+#include "benchmark.hpp"
+#include <complex>
+#include <string>
+
+namespace po = boost::program_options;
+namespace ublas = boost::numeric::ublas;
+namespace bm = boost::numeric::ublas::benchmark;
+namespace opencl = boost::numeric::ublas::opencl;
+
+namespace boost { namespace numeric { namespace ublas { namespace benchmark { namespace opencl {
+
+template <typename S, bool C> class add;
+
+template <typename V, bool C>
+class add<void(V,V,V), C> : public benchmark<void(V,V,V), C>
+{
+public:
+  add(std::string const &name) : benchmark<void(V,V,V), C>(name) {}
+  virtual void operation(long l)
+  {
+    ublas::opencl::element_add(*this->a, *this->b, *this->c, this->queue);
+  }
+};
+
+}}}}}
+
+template <typename T>
+void benchmark(std::string const &type, bool copy)
+{
+  using vector = ublas::vector<T>;
+  std::string name = "opencl::elementwise_add(vector<" + type + ">)";
+  std::vector<long> sizes({1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096});
+  if (copy)
+  {
+    bm::opencl::add<void(vector, vector, vector), true> p(name);
+    p.run(sizes);
+  }
+  else
+  {
+    bm::opencl::add<void(vector, vector, vector), false> p(name);
+    p.run(sizes);
+  }
+}
+
+int main(int argc, char **argv)
+{
+  opencl::library lib;
+  po::variables_map vm;
+  try
+  {
+    po::options_description desc("Matrix-vector product (OpenCL)\n"
+                                 "Allowed options");
+    desc.add_options()("help,h", "produce help message");
+    desc.add_options()("type,t", po::value<std::string>(), "select value-type (float, double, fcomplex, dcomplex)");
+    desc.add_options()("copy,c", po::value<bool>(), "include host<->device copy in timing");
+
+    po::store(po::parse_command_line(argc, argv, desc), vm);
+    po::notify(vm);
+
+    if (vm.count("help"))
+    {
+      std::cout << desc << std::endl;
+      return 0;
+    }
+  }
+  catch(std::exception &e)
+  {
+    std::cerr << "error: " << e.what() << std::endl;
+    return 1;
+  }
+  std::string type = vm.count("type") ? vm["type"].as<std::string>() : "float";
+  bool copy = vm.count("copy") ? vm["copy"].as<bool>() : false;
+  if (type == "float")
+    benchmark<float>("float", copy);
+  else if (type == "double")
+    benchmark<double>("double", copy);
+  else if (type == "fcomplex")
+    benchmark<std::complex<float>>("std::complex<float>", copy);
+  else if (type == "dcomplex")
+    benchmark<std::complex<double>>("std::complex<double>", copy);
+  else
+    std::cerr << "unsupported value-type \"" << vm["type"].as<std::string>() << '\"' << std::endl;
+}