csr.h

Go to the documentation of this file.

#ifndef __CSR_H__
#define __CSR_H__

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

namespace CTF_int {

  class bivar_function;

  int64_t get_csr_size(int64_t nnz, int nrow, int val_size);

  class CSR_Matrix{
    public:
      char * all_data;
      
      CSR_Matrix(int64_t nnz, int nrow, int ncol, accumulatable const * sr);

      CSR_Matrix(char * all_data);
      
      CSR_Matrix(){ all_data=NULL; }
      
      CSR_Matrix(CSR_Matrix const & other){ all_data=other.all_data; }
      
      CSR_Matrix(COO_Matrix const & coom, int nrow, int ncol, algstrct const * sr, char * data=NULL, bool init_data=true);

      int64_t nnz() const;

      int64_t size() const;

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

      char * vals() const;

      int * IA() const;

      int * JA() const;

      void partition(int s, char ** parts_buffer, CSR_Matrix ** parts);
      
      CSR_Matrix(char * const * smnds, int s);
      
      void print(algstrct const * sr);

      static void csrmm(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, bool do_offload);
      
      static void csrmultd(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, bool do_offload);

      static void csrmultcsr(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, bool do_offload);

      static void compute_has_col(

                      int const * JA,
                      int const * IA,
                      int const * JB,
                      int const * IB,
                      int         i,
                      int *       has_col);
      
      static char * csr_add(char * cA, char * cB, accumulatable const * adder);
  };
}

#endif