Grid/benchmarks/Benchmark_IO.cc

#include "Benchmark_IO.hpp"

#ifndef BENCH_IO_LMAX
#define BENCH_IO_LMAX 40
#endif

using namespace Grid;

std::string filestem(const int l)
{
  return "iobench_l" + std::to_string(l);
}

int main (int argc, char ** argv)
{
  Grid_init(&argc,&argv);

  int64_t          threads = GridThread::GetThreads();
  auto             mpi     = GridDefaultMpi();
  std::vector<int> latt;

  MSG << "Grid is setup to use " << threads << " threads" << std::endl;
  MSG << "MPI partition " << mpi << std::endl;

  MSG << SEP << std::endl;
  MSG << "Benchmark std write" << std::endl;
  MSG << SEP << std::endl;
  for (int l = 4; l <= BENCH_IO_LMAX; l += 2)
  {
    latt = {l*mpi[0], l*mpi[1], l*mpi[2], l*mpi[3]};

    MSG << "-- Local volume " << l << "^4" << std::endl;
    writeBenchmark<LatticeFermion>(latt, filestem(l), stdWrite<LatticeFermion>);
  }

  MSG << SEP << std::endl;
  MSG << "Benchmark std read" << std::endl;
  MSG << SEP << std::endl;
  for (int l = 4; l <= BENCH_IO_LMAX; l += 2)
  {
    latt = {l*mpi[0], l*mpi[1], l*mpi[2], l*mpi[3]};

    MSG << "-- Local volume " << l << "^4" << std::endl;
    readBenchmark<LatticeFermion>(latt, filestem(l), stdRead<LatticeFermion>);
  }

#ifdef HAVE_LIME
  MSG << SEP << std::endl;
  MSG << "Benchmark Grid C-Lime write" << std::endl;
  MSG << SEP << std::endl;
  for (int l = 4; l <= BENCH_IO_LMAX; l += 2)
  {
    latt = {l*mpi[0], l*mpi[1], l*mpi[2], l*mpi[3]};

    MSG << "-- Local volume " << l << "^4" << std::endl;
    writeBenchmark<LatticeFermion>(latt, filestem(l), limeWrite<LatticeFermion>);
  }

  MSG << SEP << std::endl;
  MSG << "Benchmark Grid C-Lime read" << std::endl;
  MSG << SEP << std::endl;
  for (int l = 4; l <= BENCH_IO_LMAX; l += 2)
  {
    latt = {l*mpi[0], l*mpi[1], l*mpi[2], l*mpi[3]};

    MSG << "-- Local volume " << l << "^4" << std::endl;
    readBenchmark<LatticeFermion>(latt, filestem(l), limeRead<LatticeFermion>);
  }
#endif

  Grid_finalize();

  return EXIT_SUCCESS;
}