ceph/src/boost/libs/numeric/ublas/benchmarks/mv_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/numeric/ublas/matrix.hpp>
  11 #include <boost/program_options.hpp>
  12 #include "prod.hpp"
  13 #include <complex>
  14 #include <string>
  15
  16 namespace po = boost::program_options;
  17 namespace ublas = boost::numeric::ublas;
  18 namespace bm = boost::numeric::ublas::benchmark;
  19
  20 template <typename T>
  21 void benchmark(std::string const &type)
  22 {
  23   using matrix = ublas::matrix<T, ublas::basic_row_major<>>;
  24   using vector = ublas::vector<T>;
  25   bm::prod<vector(matrix, vector)> p("prod(matrix<" + type + ">, vector<" + type + ">)");
  26   p.run(std::vector<long>({1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096}));
  27 }
  28
  29 int main(int argc, char **argv)
  30 {
  31   po::variables_map vm;
  32   try
  33   {
  34     po::options_description desc("Matrix-vector product\n"
  35                                  "Allowed options");
  36     desc.add_options()("help,h", "produce help message");
  37     desc.add_options()("type,t", po::value<std::string>(), "select value-type (float, double, fcomplex, dcomplex)");
  38
  39     po::store(po::parse_command_line(argc, argv, desc), vm);
  40     po::notify(vm);
  41
  42     if (vm.count("help"))
  43     {
  44       std::cout << desc << std::endl;
  45       return 0;
  46     }
  47   }
  48   catch(std::exception &e)
  49   {
  50     std::cerr << "error: " << e.what() << std::endl;
  51     return 1;
  52   }
  53   std::string type = vm.count("type") ? vm["type"].as<std::string>() : "float";
  54   if (type == "float")
  55     benchmark<float>("float");
  56   else if (type == "double")
  57     benchmark<double>("double");
  58   else if (type == "fcomplex")
  59     benchmark<std::complex<float>>("std::complex<float>");
  60   else if (type == "dcomplex")
  61     benchmark<std::complex<double>>("std::complex<double>");
  62   else
  63     std::cerr << "unsupported value-type \"" << vm["type"].as<std::string>() << '\"' << std::endl;
  64 }