skip.h

   1 #ifndef PET_SKIP_H
   2 #define PET_SKIP_H
   3
   4 #include <pet.h>
   5
   6 #include "state.h"
   7
   8 #if defined(__cplusplus)
   9 extern "C" {
  10 #endif
  11
  12 enum pet_skip_type {
  13         pet_skip_if,
  14         pet_skip_if_else,
  15         pet_skip_seq
  16 };
  17
  18 /* Structure that handles the construction of skip conditions.
  19  *
  20  * scop_then and scop_else represent the then and else branches
  21  *      of the if statement
  22  *
  23  * scop1 and scop2 represent the two statements that are combined
  24  *
  25  * skip[type] is true if we need to construct a skip condition of that type
  26  * equal is set if the skip conditions of types pet_skip_now and pet_skip_later
  27  *      are equal to each other
  28  * index[type] is an index expression from a zero-dimension domain
  29  *      to the virtual array representing the skip condition
  30  * scop[type] is a scop for computing the skip condition
  31  */
  32 struct pet_skip_info {
  33         isl_ctx *ctx;
  34
  35         enum pet_skip_type type;
  36
  37         int skip[2];
  38         int equal;
  39         isl_multi_pw_aff *index[2];
  40         struct pet_scop *scop[2];
  41
  42         union {
  43                 struct {
  44                         struct pet_scop *scop_then;
  45                         struct pet_scop *scop_else;
  46                 } i;
  47                 struct {
  48                         struct pet_scop *scop1;
  49                         struct pet_scop *scop2;
  50                 } s;
  51         } u;
  52 };
  53
  54 int pet_skip_info_has_skip(struct pet_skip_info *skip);
  55
  56 void pet_skip_info_if_init(struct pet_skip_info *skip, isl_ctx *ctx,
  57         struct pet_scop *scop_then, struct pet_scop *scop_else,
  58         int have_else, int affine);
  59 void pet_skip_info_if_extract_index(struct pet_skip_info *skip,
  60         __isl_keep isl_multi_pw_aff *index, struct pet_state *state);
  61 void pet_skip_info_if_extract_cond(struct pet_skip_info *skip,
  62         __isl_keep isl_pw_aff *cond, struct pet_state *state);
  63 struct pet_scop *pet_skip_info_if_add(struct pet_skip_info *skip,
  64         struct pet_scop *scop, int offset);
  65
  66 void pet_skip_info_seq_init(struct pet_skip_info *skip, isl_ctx *ctx,
  67         struct pet_scop *scop1, struct pet_scop *scop2);
  68 void pet_skip_info_seq_extract(struct pet_skip_info *skip,
  69         struct pet_state *state);
  70 struct pet_scop *pet_skip_info_seq_add(struct pet_skip_info *skip,
  71         struct pet_scop *scop, int offset);
  72
  73 #if defined(__cplusplus)
  74 }
  75 #endif
  76
  77 #endif