spctr_offload.h

Go to the documentation of this file.

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

#ifndef __SPCTR_OFFLOAD_H__
#define __SPCTR_OFFLOAD_H__

#include "../shared/offload.h"
#include "spctr_tsr.h"

namespace CTF_int {
  #ifdef OFFLOAD
  class spctr_offload : public spctr {
    public: 
      /* Class to be called on sub-blocks */
      spctr * rec_ctr;
      int iter_counter;
      int total_iter;
      int upload_phase_A;
      int upload_phase_B;
      int download_phase_C;
      int64_t size_A; /* size of A blocks */
      int64_t size_B; /* size of B blocks */
      int64_t size_C; /* size of C blocks */
      offload_arr * spr_A;
      offload_arr * spr_B;
      offload_arr * spr_C;
      
      void print();

      void run(char * A, int nblk_A, int64_t const * size_blk_A,
               char * B, int nblk_B, int64_t const * size_blk_B,
               char * C, int nblk_C, int64_t * size_blk_C,
               char *& new_C);

      int64_t spmem_fp();

      int64_t mem_rec(double nnz_frac_A, double nnz_frac_B, double nnz_frac_C);

      double est_time_fp(int nlyr, double nnz_frac_A, double nnz_frac_B, double nnz_frac_C);

      double est_time_rec(int nlyr, double nnz_frac_A, double nnz_frac_B, double nnz_frac_C);

      spctr * clone();

      spctr_offload(spctr * other);

      ~spctr_offload();

      spctr_offload(contraction const * c,
                    int64_t size_A,
                    int64_t size_B,
                    int64_t size_C,
                    int total_iter,
                    int upload_phase_A,
                    int upload_phase_B,
                    int download_phase_C);

  };
  #endif

}
#endif