strp_tsr.h

Go to the documentation of this file.

/*Copyright (c) 2011, Edgar Solomonik, all rights reserved.*/

#ifndef __STRP_TSR_H__
#define __STRP_TSR_H__

#include "../scaling/scale_tsr.h"
#include "../summation/sum_tsr.h"
#include "../contraction/ctr_comm.h"
#include "../mapping/mapping.h"

namespace CTF_int {
  class summation;

  class strp_tsr {
    public: 
      int              alloced;
      int              order;
      int64_t          blk_sz;
      int *            edge_len;
      int *            strip_dim;
      int *            strip_idx;
      char *           A;
      char *           buffer;
      algstrct const * sr_A;
      
      void run(int const dir);

      void free_exp();

      int64_t mem_fp();

      strp_tsr * clone();

      strp_tsr(strp_tsr * o);
      ~strp_tsr(){ if (buffer != NULL) CTF_int::cdealloc(buffer); CTF_int::cdealloc(edge_len); CTF_int::cdealloc(strip_dim); CTF_int::cdealloc(strip_idx);}
      strp_tsr(){ buffer = NULL; }
  };

  class strp_scl : public scl {
    public: 
      scl * rec_scl;
      
      strp_tsr * rec_strp;

      void run();

      int64_t mem_fp();

      scl * clone();
      
      strp_scl(scl * other);

      ~strp_scl();
      strp_scl(){}
  };

  class strp_sum : public tsum {
    public: 
      tsum * rec_tsum;
      
      strp_tsr * rec_strp_A;
      strp_tsr * rec_strp_B;

      int strip_A;
      int strip_B;
      
      void run();

      int64_t mem_fp();

      tsum * clone();
      
      strp_sum(tsum * other);

      ~strp_sum();
      strp_sum(summation const * s);
  };

  class strp_ctr : public ctr {
    public: 
      ctr * rec_ctr;
      
      strp_tsr * rec_strp_A;
      strp_tsr * rec_strp_B;
      strp_tsr * rec_strp_C;

      int strip_A;
      int strip_B;
      int strip_C;
      
      void run(char * A, char * B, char * C);

      int64_t mem_fp();
      int64_t mem_rec();


      double est_time_rec(int nlyr);
  
      ctr * clone();
    
      ~strp_ctr();

      strp_ctr(ctr *other);
      strp_ctr(contraction const * c) : ctr(c) {}
  };

  int strip_diag(int              order,
                 int              order_tot,
                 int const *      idx_map,
                 int64_t          vrt_sz,
                 mapping const *  edge_map,
                 topology const * topo,
                 algstrct const * sr,
                 int *            blk_edge_len,
                 int64_t *        blk_sz,
                 strp_tsr **      stpr);

}

#endif // __STRP_TSR_H__