nvgraph_triangle_counter.hpp

Go to the documentation of this file.

#pragma once

#include <nvgraph.h>

#include "pangolin/triangle_counter/triangle_counter.hpp"
#include "pangolin/dag_lowertriangular_csr.hpp"

PANGOLIN_BEGIN_NAMESPACE()

class NvGraphTriangleCounter : public TriangleCounter
{
private:
  int gpu_;
  DAGLowerTriangularCSR dag_;
  nvgraphCSRTopology32I_t csr_;

public:
  NvGraphTriangleCounter(Config &c);
  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 dag_.num_edges(); }
};

PANGOLIN_END_NAMESPACE()