spmm_tc.hpp

Go to the documentation of this file.

#pragma once

#include "pangolin/triangle_counter/cuda_triangle_counter.hpp"
#include "pangolin/sparse/gpu_csr.hpp"

#include <iostream>

PANGOLIN_BEGIN_NAMESPACE()


class SpmmTC : public CUDATriangleCounter
{
  private:
    uint64_t *edgeCnt_; 
    Uint *edgeSrc_; 
    Uint *edgeDst_; 
    pangolin::GPUCSR<Uint> aL_; 
    pangolin::GPUCSR<Uint> aU_; 
    uint64_t *nextEdge_; // next edge that GPU counter should write an edge count to

  public:
    SpmmTC(Config &c);
    virtual ~SpmmTC();
    virtual void read_data(const std::string &path) override;
    virtual void setup_data() override;
    virtual size_t count() override;
    virtual uint64_t num_edges() override { return aL_.nnz(); }
    virtual size_t num_nodes() override { return aL_.num_nodes(); }
};

PANGOLIN_END_NAMESPACE()