ceph/src/boost/libs/numeric/ublas/benchmarks/reference/outer_prod.cpp

   1 //
   2 // Copyright (c) 2018 Stefan Seefeld
   3 // All rights reserved.
   4 //
   5 // This file is part of Boost.uBLAS. It is made available under the
   6 // Boost Software License, Version 1.0.
   7 // (Consult LICENSE or http://www.boost.org/LICENSE_1_0.txt)
   8
   9 #include <boost/numeric/ublas/vector.hpp>
  10 #include <boost/program_options.hpp>
  11 #include "../init.hpp"
  12 #include "../benchmark.hpp"
  13 #include <complex>
  14 #include <string>
  15
  16 namespace boost { namespace numeric { namespace ublas { namespace benchmark {
  17
  18 template <typename S> class outer_prod;
  19
  20 template <typename R, typename V1, typename V2>
  21 class outer_prod<R(V1, V2)> : public benchmark
  22 {
  23 public:
  24   outer_prod(std::string const &name) : benchmark(name) {}
  25   virtual void setup(long l)
  26   {
  27     init(a, l, 200);
  28     init(b, l, 200);
  29   }
  30   virtual void operation(long l)
  31   {
  32     for (int i = 0; i < l; ++i)
  33       for (int j = 0; j < l; ++j)
  34         c(i,j) = - a(i) * b(j);
  35   }
  36 private:
  37   V1 a;
  38   V2 b;
  39   R c;
  40 };
  41
  42 }}}}
  43
  44 namespace po = boost::program_options;
  45 namespace ublas = boost::numeric::ublas;
  46 namespace bm = boost::numeric::ublas::benchmark;
  47
  48 template <typename T>
  49 void benchmark(std::string const &type)
  50 {
  51   using vector = ublas::vector<T>;
  52   using matrix = ublas::matrix<T>;
  53   bm::outer_prod<matrix(vector, vector)> p("ref::outer_prod(vector<" + type + ">)");
  54   p.run(std::vector<long>({1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096}));
  55 }
  56
  57 int main(int argc, char **argv)
  58 {
  59   po::variables_map vm;
  60   try
  61   {
  62     po::options_description desc("Outer product (reference implementation)\n"
  63                                  "Allowed options");
  64     desc.add_options()("help,h", "produce help message");
  65     desc.add_options()("type,t", po::value<std::string>(), "select value-type (float, double, fcomplex, dcomplex)");
  66
  67     po::store(po::parse_command_line(argc, argv, desc), vm);
  68     po::notify(vm);
  69
  70     if (vm.count("help"))
  71     {
  72       std::cout << desc << std::endl;
  73       return 0;
  74     }
  75   }
  76   catch(std::exception &e)
  77   {
  78     std::cerr << "error: " << e.what() << std::endl;
  79     return 1;
  80   }
  81   std::string type = vm.count("type") ? vm["type"].as<std::string>() : "float";
  82   if (type == "float")
  83     benchmark<float>("float");
  84   else if (type == "double")
  85     benchmark<double>("double");
  86   else if (type == "fcomplex")
  87     benchmark<std::complex<float>>("std::complex<float>");
  88   else if (type == "dcomplex")
  89     benchmark<std::complex<double>>("std::complex<double>");
  90   else
  91     std::cerr << "unsupported value-type \"" << vm["type"].as<std::string>() << '\"' << std::endl;
  92 }