ublas/benchmarks/inner_prod.cpp
2018-09-28 19:23:20 -04:00

90 lines
2.4 KiB
C++

//
// Copyright (c) 2018 Stefan Seefeld
// All rights reserved.
//
// This file is part of Boost.uBLAS. It is made available under the
// Boost Software License, Version 1.0.
// (Consult LICENSE or http://www.boost.org/LICENSE_1_0.txt)
#include <boost/numeric/ublas/vector.hpp>
#include <boost/program_options.hpp>
#include "init.hpp"
#include "benchmark.hpp"
#include <complex>
#include <string>
namespace boost { namespace numeric { namespace ublas { namespace benchmark {
template <typename S> class inner_prod;
template <typename R, typename V1, typename V2>
class inner_prod<R(V1, V2)> : public benchmark
{
public:
inner_prod(std::string const &name) : benchmark(name) {}
virtual void setup(long l)
{
init(a, l, 200);
init(b, l, 200);
}
virtual void operation(long l)
{
c = ublas::inner_prod(a, b);
}
private:
V1 a;
V2 b;
R c;
};
}}}}
namespace po = boost::program_options;
namespace ublas = boost::numeric::ublas;
namespace bm = boost::numeric::ublas::benchmark;
template <typename T>
void benchmark(std::string const &type)
{
using vector = ublas::vector<T>;
bm::inner_prod<T(vector, vector)> p("inner_prod(vector<" + type + ">)");
p.run(std::vector<long>({1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536}));
}
int main(int argc, char **argv)
{
po::variables_map vm;
try
{
po::options_description desc("Inner product\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)");
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";
if (type == "float")
benchmark<float>("float");
else if (type == "double")
benchmark<double>("double");
else if (type == "fcomplex")
benchmark<std::complex<float>>("std::complex<float>");
else if (type == "dcomplex")
benchmark<std::complex<double>>("std::complex<double>");
else
std::cerr << "unsupported value-type \"" << vm["type"].as<std::string>() << '\"' << std::endl;
}