spmv.h

Go to the documentation of this file.

#ifndef SPIKE_SPMV_H
#define SPIKE_SPMV_H


#include <cusp/multiply.h>

#include <spike/timer.h>


namespace spike {



template <typename Matrix>
class SpmvCusp {
public:
    SpmvCusp(Matrix& A) : m_A(A), m_time(0), m_count(0) {}

    double getTime() const   {return m_time;}

    double getCount() const  {return m_count;}

    double getGFlops() const
    {
        double avgTime = m_time / m_count;
        return 2 * m_A.num_entries / (1e6 * avgTime);
    }

    template <typename Array>
    void operator()(const Array& v,
                    Array&       Av)
    {
        m_timer.Start();
        cusp::multiply(m_A, v, Av);
        m_timer.Stop();

        m_count++;
        m_time += m_timer.getElapsed();
    }

private:
    Matrix&      m_A;
    GPUTimer     m_timer;
    double       m_time;
    int          m_count;
};

} // namespace spike


#endif