root / tests / benchfgemm.C

//#include "goto-def.h"

#include <iostream>

#include "fflas-ffpack/fflas.h"
#include "fflas-ffpack/modular-balanced.h"
#include "timer.h"
#include "Matio.h"

using namespace std;

int main(int argc, char** argv) {

  // parameter: p, n, iteration, file1, file2

  double    p    = atof(argv[1]);
  int n    = atoi(argv[2]);
  size_t w = atoi (argv[3]);
  size_t iter = atoi(argv[4]);

  //typedef ModularBalanced<double> Field;
  typedef ModularBalanced<float> Field;
  typedef Field::Element Element;
      
  Field F(p);
  Element one,zero;
  F.init(one, 1.0);
  F.init(zero,0.0);

  Timer chrono;
  double time=0.0, time2=0.0;
  int singular;
  
  Element * A, * B, * C;

  for (size_t i=0;i<iter;++i){
       
    Field::RandIter G(F);
    A = new Element[n*n];
    for (size_t i=0; i<n*n; ++i)
      G.random (*(A+i));      

    B = new Element[n*n];
    for (size_t i=0; i<n*n; ++i)
      G.random(*(B+i));

    C = new Element[n*n];

    chrono.clear();
    chrono.start();
    FFLAS::fgemm (F, FFLAS::FflasNoTrans, FFLAS::FflasNoTrans, n,n,n, one,
                  A, n, B, n, zero, C,n, w);
    chrono.stop();
    time+=chrono.realtime();

    delete[] A;
    delete[] B;
    delete[] C;
  }
  
  std::cerr<<"n: "<<n <<" p: "<<p<<" w: "<<w<<std::endl
           <<" time:  "<<time/iter<<" s"<<std::endl
           <<" speed: "<<2.0*n/1000.0*n/1000.0/time*n/1000.0*iter<<" Gffops"
           <<std::endl;

  return 0;
}