coo.h

Go to the documentation of this file.

#ifndef __COO_H__
#define __COO_H__

#include "../tensor/algstrct.h"

namespace CTF_int {

  class CSR_Matrix;
  class bivar_function;

  int64_t get_coo_size(int64_t nnz, int val_size);

  class COO_Matrix{
    public:
      char * all_data;
      
      COO_Matrix(int64_t nnz, algstrct const * sr);

      COO_Matrix(char * all_data);

      COO_Matrix(CSR_Matrix const & csr, algstrct const * sr);

      int64_t nnz() const;

      int64_t size() const;

      int nrow() const;
      
      int ncol() const;
      
      int val_size() const;

      char * vals() const;

      int * rows() const;

      int * cols() const;

      void set_data(int64_t nz, int order, int const * lens, int const * ordering, int nrow_idx, char const * tsr_data, algstrct const * sr, int const * phase);

      void get_data(int64_t nz, int order, int const * lens, int const * rev_ordering, int nrow_idx, char * tsr_data, algstrct const * sr, int const * phase, int const * phase_rank);

      static void coomm(char const * A, algstrct const * sr_A, int m, int n, int k, char const * alpha, char const * B, algstrct const * sr_B, char const * beta, char * C, algstrct const * sr_C, bivar_function const * func);

  };
}

#endif