distribution.h

Go to the documentation of this file.

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

#ifndef __INT_DISTRIBUTION_H__
#define __INT_DISTRIBUTION_H__

#include "mapping.h"

namespace CTF_int {

  class tensor;

  inline
  int get_distribution_size(int order){
    return sizeof(int)*2 + sizeof(int64_t) + order*sizeof(int)*7;
  }

  // \brief data distribution object used for redistribution
  class distribution {
    public:
      int     order;
      int *   phase;
      int *   virt_phase;
      int *   phys_phase;
      int *   pe_lda;
      int *   pad_edge_len;
      int *   padding;
      int *   perank;
      int     is_cyclic;
      int64_t size;

      distribution();
      ~distribution();

      distribution(tensor const * tsr);

      distribution(char const * buffer);

      void serialize(char ** buffer, int * size);
    private:
      void free_data();
  };

  void calc_dim(int             order,
                int64_t         size,
                int const *     edge_len,
                mapping const * edge_map,
                int64_t *       vrt_sz,
                int *           vrt_edge_len,
                int *           blk_edge_len);
}

#endif