isl_bind_domain_templ.c

Go to the documentation of this file.

/*
 * Copyright 2018      Cerebras Systems
 *
 * Use of this software is governed by the MIT license
 *
 * Written by Sven Verdoolaege,
 * Cerebras Systems, 175 S San Antonio Rd, Los Altos, CA, USA
 */
 
#include <isl/space.h>
 
/* Merge parameter "param" into the input dimension "i" of "obj".
 *
 * First plug in the parameter for the input dimension in "obj".
 * The drop the (now defunct) input dimension and
 * move the parameter in its original position.
 * Since dimension manipulations destroy spaces, modify the space
 * separately by only dropping the parameter.
 */
static __isl_give TYPE *FN(TYPE,merge_param)(__isl_take TYPE *obj, int i,
  int param)
{
  isl_id *id;
  isl_aff *aff;
  isl_space *space;
  isl_multi_aff *ma;
 
  space = FN(TYPE,get_domain_space)(obj);
  id = isl_space_get_dim_id(space, isl_dim_param, param);
  aff = isl_aff_param_on_domain_space_id(isl_space_copy(space), id);
  space = isl_space_map_from_set(space);
  ma = isl_multi_aff_identity(space);
  ma = isl_multi_aff_set_aff(ma, i, aff);
  obj = FN(TYPE,pullback_multi_aff)(obj, ma);
  space = FN(TYPE,get_domain_space)(obj);
  obj = FN(TYPE,drop_dims)(obj, isl_dim_in, i, 1);
  obj = FN(TYPE,move_dims)(obj, isl_dim_in, i, isl_dim_param, param, 1);
  space = isl_space_drop_dims(space, isl_dim_param, param, 1);
  obj = FN(TYPE,reset_domain_space)(obj, space);
 
  return obj;
}
 
/* Given a tuple of identifiers "tuple" that correspond
 * to the initial input dimensions of "obj",
 * if any of those identifiers appear as parameters
 * in "obj", then equate those parameters with the corresponding
 * input dimensions and project out the parameters.
 * The result therefore has no such parameters.
 */
static __isl_give TYPE *FN(TYPE,equate_initial_params)(__isl_take TYPE *obj,
  __isl_keep isl_multi_id *tuple)
{
  int i;
  isl_size n;
 
  n = isl_multi_id_size(tuple);
  if (n < 0)
    return FN(TYPE,free)(obj);
  for (i = 0; i < n; ++i) {
    isl_id *id;
    int pos;
 
    id = isl_multi_id_get_at(tuple, i);
    if (!id)
      return FN(TYPE,free)(obj);
    pos = FN(TYPE,find_dim_by_id)(obj, isl_dim_param, id);
    isl_id_free(id);
    if (pos < 0)
      continue;
    obj = FN(TYPE,merge_param)(obj, i, pos);
  }
 
  return obj;
}
 
/* Given a tuple of identifiers "tuple" in a space that corresponds
 * to the domain of "obj", if any of those identifiers appear as parameters
 * in "obj", then equate those parameters with the corresponding
 * input dimensions and project out the parameters.
 * The result therefore has no such parameters.
 */
static __isl_give TYPE *FN(TYPE,equate_domain_params)(__isl_take TYPE *obj,
  __isl_keep isl_multi_id *tuple)
{
  isl_stat r;
  isl_space *obj_space, *tuple_space;
 
  obj_space = FN(TYPE,get_space)(obj);
  tuple_space = isl_multi_id_peek_space(tuple);
  r = isl_space_check_domain_tuples(tuple_space, obj_space);
  isl_space_free(obj_space);
  if (r < 0)
    return FN(TYPE,free)(obj);
 
  return FN(TYPE,equate_initial_params)(obj, tuple);
}
 
/* Bind the domain dimensions of the function "obj" to parameters
 * with identifiers specified by "tuple", living in the same space
 * as the domain of "obj".
 *
 * If no parameters with these identifiers appear in "obj" already,
 * then the domain dimensions are simply reinterpreted as parameters.
 * Otherwise, the parameters are first equated to the corresponding
 * domain dimensions.
 */
__isl_give TYPE *FN(TYPE,bind_domain)(__isl_take TYPE *obj,
  __isl_take isl_multi_id *tuple)
{
  isl_space *space;
 
  obj = FN(TYPE,equate_domain_params)(obj, tuple);
  space = FN(TYPE,get_space)(obj);
  space = isl_space_bind_map_domain(space, tuple);
  isl_multi_id_free(tuple);
  obj = FN(TYPE,reset_space)(obj, space);
 
  return obj;
}
 
/* Given a tuple of identifiers "tuple" in a space that corresponds
 * to the domain of the wrapped relation in the domain of "obj",
 * if any of those identifiers appear as parameters
 * in "obj", then equate those parameters with the corresponding
 * input dimensions and project out the parameters.
 * The result therefore has no such parameters.
 */
static __isl_give TYPE *FN(TYPE,equate_domain_wrapped_domain_params)(
  __isl_take TYPE *obj, __isl_keep isl_multi_id *tuple)
{
  isl_stat r;
  isl_space *obj_space, *tuple_space;
 
  obj_space = FN(TYPE,get_space)(obj);
  tuple_space = isl_multi_id_peek_space(tuple);
  r = isl_space_check_domain_wrapped_domain_tuples(tuple_space,
              obj_space);
  isl_space_free(obj_space);
  if (r < 0)
    return FN(TYPE,free)(obj);
 
  return FN(TYPE,equate_initial_params)(obj, tuple);
}
 
/* Given a function living in a space of the form [A -> B] -> C and
 * a tuple of identifiers in A, bind the domain dimensions of the relation
 * wrapped in the domain of "obj" with identifiers specified by "tuple",
 * returning a function in the space B -> C.
 *
 * If no parameters with these identifiers appear in "obj" already,
 * then the domain dimensions are simply reinterpreted as parameters.
 * Otherwise, the parameters are first equated to the corresponding
 * domain dimensions.
 */
__isl_give TYPE *FN(TYPE,bind_domain_wrapped_domain)(__isl_take TYPE *obj,
  __isl_take isl_multi_id *tuple)
{
  isl_space *space;
 
  obj = FN(TYPE,equate_domain_wrapped_domain_params)(obj, tuple);
  space = FN(TYPE,get_space)(obj);
  space = isl_space_bind_domain_wrapped_domain(space, tuple);
  isl_multi_id_free(tuple);
  obj = FN(TYPE,reset_space)(obj, space);
 
  return obj;
}