isl_map_bound_templ.c

   1 /*
   2  * Copyright 2018      Cerebras Systems
   3  *
   4  * Use of this software is governed by the MIT license
   5  *
   6  * Written by Sven Verdoolaege,
   7  * Cerebras Systems, 175 S San Antonio Rd, Los Altos, CA, USA
   8  */
   9
  10 #include "isl_multi_macro.h"
  11 #undef TYPE
  12 #define TYPE CAT(isl_,BASE)
  13
  14 /* Check that "map" and "multi" live in the same space, ignoring parameters.
  15  */
  16 static isl_stat FN(check_map_equal_tuples_multi,BASE)(__isl_keep isl_map *map,
  17         __isl_keep MULTI(BASE) *multi)
  18 {
  19         isl_space *map_space, *multi_space;
  20
  21         map_space = isl_map_peek_space(map);
  22         multi_space = FN(MULTI(BASE),peek_space)(multi);
  23         return isl_space_check_equal_tuples(map_space, multi_space);
  24 }
  25
  26 /* Apply "map_bound" to "map" with the corresponding value in "bound"
  27  * for each output dimension.
  28  * If "bound" has an explicit domain (which implies that "bound"
  29  * is zero-dimensional), then intersect the domain of "map"
  30  * with this explicit domain instead.
  31  */
  32 static __isl_give isl_map *FN(map_bound_multi,BASE)(__isl_take isl_map *map,
  33         __isl_take MULTI(BASE) *bound,
  34         __isl_give isl_map *map_bound(__isl_take isl_map *map,
  35                 unsigned pos, __isl_take TYPE *value))
  36 {
  37         int i;
  38         isl_size dim;
  39
  40         dim = isl_map_dim(map, isl_dim_out);
  41         if (dim < 0 || FN(check_map_equal_tuples_multi,BASE)(map, bound) < 0)
  42                 goto error;
  43
  44         for (i = 0; i < dim; ++i) {
  45                 TYPE *el;
  46
  47                 el = FN(MULTI(BASE),get_at)(bound, i);
  48                 map = map_bound(map, i, el);
  49         }
  50         map = FN(FN(isl_map_intersect_multi,BASE),explicit_domain)(map, bound);
  51         FN(MULTI(BASE),free)(bound);
  52         return map;
  53 error:
  54         isl_map_free(map);
  55         FN(MULTI(BASE),free)(bound);
  56         return NULL;
  57 }