qdp_flopcount.h

Go to the documentation of this file.

// -*- C++ -*-
 
#ifndef QDP_FLOPCOUNT_H
#define QDP_FLOPCOUNT_H
 
namespace QDP
{
 
  //---------------------------------------------------------------------

  class FlopCounter {
  public:
    FlopCounter(void) : count(0), sitesOnNode((unsigned long long)Layout::sitesOnNode()) {}

    ~FlopCounter() {} 

    FlopCounter(const FlopCounter& c) : count(c.count), sitesOnNode(c.sitesOnNode) {}

    inline void reset(void) { 
      count = 0;
    }

    inline void addFlops(unsigned long long flops) { 
      count += flops;
    }

    inline void addSiteFlops(unsigned long  long flops) { 
      count += (flops * sitesOnNode);
    }

    inline void addSiteFlops(unsigned long flops, const Subset& s) {
      count += (flops * (unsigned long long)(s.numSiteTable()));
    }

    inline unsigned long long getFlops(void) const { 
      return count;
    }

    inline void report(const std::string& name, 
                       const double& time_in_seconds) {
 
      double mflops_per_cpu = (double)count/((double)(1000*1000)*time_in_seconds);
      double mflops_overall = mflops_per_cpu;
      QDPInternal::globalSum(mflops_overall);
      double gflops_overall = mflops_overall/(double)(1000);
      double tflops_overall = gflops_overall/(double)(1000);
 
      QDPIO::cout <<"QDP:FlopCount:" << name << " Performance/CPU: t=" << time_in_seconds << "(s) Flops=" << (double)count << " => " << mflops_per_cpu << " Mflops/cpu." << std::endl;
      QDPIO::cout << "QDP:FlopCount:"  << name <<" Total performance:  " << mflops_overall << " Mflops = " << gflops_overall << " Gflops = " << tflops_overall << " Tflops" << std::endl;
    }
 
  private:
    unsigned long long count;
    const  unsigned long long sitesOnNode;
  };
  // end of group 
 
} // namespace QDP
 
#endif