48 lines · c
1/*2 * Copyright 2019 Cerebras Systems3 *4 * Use of this software is governed by the MIT license5 *6 * Written by Sven Verdoolaege,7 * Cerebras Systems, 175 S San Antonio Rd, Los Altos, CA, USA8 */9 10#include <isl_pw_macro.h>11 12#undef VAL13#define VAL CAT(isl_,VAL_BASE)14 15/* Add "v" to the constant term of "pw" over its entire definition domain.16 */17__isl_give PW *FN(FN(PW,add_constant),VAL_BASE)(__isl_take PW *pw,18 __isl_take VAL *v)19{20 isl_bool zero;21 isl_size n;22 int i;23 24 zero = FN(VAL,is_zero)(v);25 n = FN(PW,n_piece)(pw);26 if (zero < 0 || n < 0)27 goto error;28 if (zero || n == 0) {29 FN(VAL,free)(v);30 return pw;31 }32 33 for (i = 0; i < n; ++i) {34 EL *el;35 36 el = FN(PW,take_base_at)(pw, i);37 el = FN(FN(EL,add_constant),VAL_BASE)(el, FN(VAL,copy)(v));38 pw = FN(PW,restore_base_at)(pw, i, el);39 }40 41 FN(VAL,free)(v);42 return pw;43error:44 FN(PW,free)(pw);45 FN(VAL,free)(v);46 return NULL;47}48