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 */
FN(check_map_equal_tuples_multi,BASE)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 */
FN(map_bound_multi,BASE)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 }
58