mapping.h

Go to the documentation of this file.

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

#ifndef __INT_MAPPING_H__
#define __INT_MAPPING_H__

#include "topology.h"

namespace CTF_int {

  class CommData;
  class tensor;

  enum map_type {
    NOT_MAPPED,
    PHYSICAL_MAP,
    VIRTUAL_MAP
  };

  // \brief object keeping track of a mapping of a tensor dimension onto a torus dimension
  class mapping {
    public:
      map_type type;
      int np;
      int cdt;
      int has_child;
      mapping * child;

      mapping();

      ~mapping();

      mapping(mapping const & other);
      mapping & operator=(mapping const & other);
  
      int calc_phase() const;
      
      int calc_phys_phase() const;
      
      int calc_phys_rank(topology const * topo) const;

      void clear();

      void aug_phys(topology const * topo, int idim);
      
      void aug_virt(int tot_phase);
  };
  
  int comp_dim_map(mapping const * map_A,
                   mapping const * map_B);

  void copy_mapping(int             order,
                    mapping const * mapping_A,
                    mapping *       mapping_B);

  int copy_mapping(int             order_A,
                   int             order_B,
                   int const *     idx_A,
                   mapping const * mapping_A,
                   int const *     idx_B,
                   mapping *       mapping_B,
                   int             make_virt = 1);


  int map_tensor(int         num_phys_dims,
                 int         tsr_order,
                 int const * tsr_edge_len,
                 int const * tsr_sym_table,
                 int *       restricted,
                 CommData  * phys_comm,
                 int const * comm_idx,
                 int         fill,
                 mapping *   tsr_edge_map);

  int check_self_mapping(tensor const * tsr,
                         int const *    idx_map);

  int map_self_indices(tensor const * tsr,
                       int const *    idx_map);


  int map_symtsr(int         tsr_order,
                 int const * tsr_sym_table,
                 mapping *   tsr_edge_map);

  
  int stretch_virt(int       order,
                   int       stretch_factor,
                   mapping * maps);


}

#endif