void clay_relation_substitute(osl_relation_p relation,
int original_col,
int target_col,
int factor) {
int row;
osl_int_t tmp;
// Do not allow substituting e/i flag or the constant and
// do not allow substituting by e/i flag (but allow by constant).
if (original_col <= 0 || original_col > relation->nb_columns - 1 ||
target_col <= 0 || target_col > relation->nb_columns)
return;
osl_int_init(relation->precision, &tmp);
for (row = 0; row < relation->nb_rows; row++) {
if (osl_int_zero(relation->precision, relation->m[row][original_col])) {
continue;
}
osl_int_mul_si(relation->precision, &tmp, relation->m[row][original_col],
factor);
osl_int_add(relation->precision, &relation->m[row][target_col],
relation->m[row][target_col], tmp);
}
osl_int_clear(relation->precision, &tmp);
}
/**
* clay_util_statement_set_vector function:
* Set the equation on each line where the column of the output dim is
* different of zero
* \param[in,out] statement
* \param[in] vector {(([output, ...],) [param, ..],) [const]}
* \param[in] column column on the output dim
*/
void clay_util_statement_set_vector(
osl_statement_p statement,
clay_list_p vector, int column) {
osl_relation_p scattering = statement->scattering;
clay_array_p arr_dims = NULL, arr_params = NULL, arr_const = NULL;
int i, j, k;
int precision = scattering->precision;
osl_int_p tmp;
tmp = osl_int_malloc(precision);
if (vector->size > 3) {
CLAY_error("list with more than 3 arrays not supported");
} else if (vector->size == 3) {
arr_dims = vector->data[0];
arr_params = vector->data[1];
arr_const = vector->data[2];
} else if (vector->size == 2) {
arr_params = vector->data[0];
arr_const = vector->data[1];
} else {
arr_const = vector->data[0];
}
// for each line where there is a number different from zero on the
// column
for (k = 0 ; k < scattering->nb_rows ; k++) {
if (!osl_int_zero(precision, scattering->m[k][1+column])) {
// scattering = coeff_outputdim * shifting
// affect output dims
if (vector->size >= 3) {
i = 1;
for (j = 0 ; j < arr_dims->size ; j++) {
osl_int_mul_si(precision,
&scattering->m[k][i],
scattering->m[k][1+column],
arr_dims->data[j]);
i++;
}
}
// here we add we the last value
// scattering += coeff_outputdim * shifting
// affects parameters
if (vector->size >= 2) {
i = 1 + scattering->nb_output_dims + scattering->nb_input_dims +
scattering->nb_local_dims;
for (j = 0 ; j < arr_params->size ; j++) {
osl_int_mul_si(precision,
tmp,
scattering->m[k][1+column],
arr_params->data[j]);
osl_int_add(precision,
&scattering->m[k][i],
scattering->m[k][i],
*tmp);
i++;
}
}
// set the constant
if (vector->size >= 1 && arr_const->size == 1) {
osl_int_mul_si(precision,
tmp,
scattering->m[k][1+column],
arr_const->data[0]);
osl_int_add(precision,
&scattering->m[k][scattering->nb_columns-1],
scattering->m[k][scattering->nb_columns-1],
*tmp);
}
}
}
osl_int_free(precision, tmp);
}
int main(int argc, char** argv)
{
if (argc > 1) { printf("argv are ignored\n"); }
unsigned int nb_fail = 0;
#ifdef OSL_GMP_IS_HERE
int i;
for (i = SCHAR_MIN; i <= SCHAR_MAX; ++i) {
osl_int_t a_sp; osl_int_init_set_si(OSL_PRECISION_SP, &a_sp, i);
osl_int_t a_dp; osl_int_init_set_si(OSL_PRECISION_DP, &a_dp, i);
osl_int_t a_mp; osl_int_init_set_si(OSL_PRECISION_MP, &a_mp, i);
int j;
for (j = SCHAR_MIN; j <= SCHAR_MAX; ++j) {
int error = 0;
osl_int_t b_sp; osl_int_init_set_si(OSL_PRECISION_SP, &b_sp, j);
osl_int_t b_dp; osl_int_init_set_si(OSL_PRECISION_DP, &b_dp, j);
osl_int_t b_mp; osl_int_init_set_si(OSL_PRECISION_MP, &b_mp, j);
osl_int_t c_sp; osl_int_init(OSL_PRECISION_SP, &c_sp);
osl_int_t c_dp; osl_int_init(OSL_PRECISION_DP, &c_dp);
osl_int_t c_mp; osl_int_init(OSL_PRECISION_MP, &c_mp);
int const a_sp_i = osl_int_get_si(OSL_PRECISION_SP, a_sp);
int const a_dp_i = osl_int_get_si(OSL_PRECISION_DP, a_dp);
int const a_mp_i = osl_int_get_si(OSL_PRECISION_MP, a_mp);
int const b_sp_i = osl_int_get_si(OSL_PRECISION_SP, b_sp);
int const b_dp_i = osl_int_get_si(OSL_PRECISION_DP, b_dp);
int const b_mp_i = osl_int_get_si(OSL_PRECISION_MP, b_mp);
// osl_int_init_set_si & osl_int_init & osl_int_get_si
if (!error) {
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (a_sp_i != a_dp_i || a_sp_i != a_mp_i) {
error++; printf("Error osl_int_init_set_si or osl_int_get_si\n");
}
if (b_sp_i != b_dp_i || b_sp_i != b_mp_i) {
error++; printf("Error osl_int_init_set_si or osl_int_get_si\n");
}
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != 0) {
error++; printf("Error osl_int_init or osl_int_get_si\n");
}
}
// osl_int_assign
if (!error) {
osl_int_assign(OSL_PRECISION_SP, &c_sp, b_sp);
osl_int_assign(OSL_PRECISION_DP, &c_dp, b_dp);
osl_int_assign(OSL_PRECISION_MP, &c_mp, b_mp);
if (osl_int_ne(OSL_PRECISION_SP, c_sp, b_sp)) {
error++; printf("Error osl_int_assign\n");
}
if (osl_int_ne(OSL_PRECISION_DP, c_dp, b_dp)) {
error++; printf("Error osl_int_assign\n");
}
if (osl_int_ne(OSL_PRECISION_MP, c_mp, b_mp)) {
error++; printf("Error osl_int_assign\n");
}
}
// osl_int_swap
// osl_int_increment
if (!error) {
osl_int_increment(OSL_PRECISION_SP, &c_sp, a_sp);
osl_int_increment(OSL_PRECISION_DP, &c_dp, a_dp);
osl_int_increment(OSL_PRECISION_MP, &c_mp, a_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != a_sp_i + 1) {
error++; printf("Error osl_int_increment\n");
}
}
// osl_int_decrement
if (!error) {
osl_int_decrement(OSL_PRECISION_SP, &c_sp, a_sp);
osl_int_decrement(OSL_PRECISION_DP, &c_dp, a_dp);
osl_int_decrement(OSL_PRECISION_MP, &c_mp, a_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != a_sp_i - 1) {
error++; printf("Error osl_int_decrement\n");
}
}
// osl_int_add
if (!error) {
osl_int_add(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_add(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_add(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != a_sp_i + b_sp_i) {
error++; printf("Error osl_int_add\n");
}
}
// osl_int_add_si
// osl_int_sub
if (!error) {
osl_int_sub(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_sub(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_sub(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != a_sp_i - b_sp_i) {
error++; printf("Error osl_int_add\n");
}
}
// osl_int_mul
if (!error) {
osl_int_mul(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_mul(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_mul(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != a_sp_i * b_sp_i) {
error++; printf("Error osl_int_mul\n");
}
}
// osl_int_mul_si
if (!error) {
osl_int_mul_si(OSL_PRECISION_SP, &c_sp, a_sp, b_sp_i);
osl_int_mul_si(OSL_PRECISION_DP, &c_dp, a_dp, b_dp_i);
osl_int_mul_si(OSL_PRECISION_MP, &c_mp, a_mp, b_mp_i);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != a_sp_i * b_sp_i) {
error++; printf("Error osl_int_mul_si\n");
}
}
// osl_int_div_exact
if (!error && b_sp_i != 0 && a_sp_i % b_sp_i == 0) {
osl_int_div_exact(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_div_exact(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_div_exact(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i || c_sp_i != a_sp_i / b_sp_i) {
error++; printf("Error osl_int_div_exact\n");
}
}
// osl_int_floor_div_q
if (!error && b_sp_i != 0) {
osl_int_floor_div_q(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_floor_div_q(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_floor_div_q(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i) {
error++; printf("Error osl_int_floor_div_q\n");
}
}
// osl_int_floor_div_r
if (!error && b_sp_i != 0) {
osl_int_floor_div_r(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_floor_div_r(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_floor_div_r(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i) {
error++; printf("Error osl_int_floor_div_r\n");
}
}
// osl_int_floor_div_q_r
if (!error && b_sp_i != 0) {
osl_int_t q_sp; osl_int_init(OSL_PRECISION_SP, &q_sp);
osl_int_t q_dp; osl_int_init(OSL_PRECISION_DP, &q_dp);
osl_int_t q_mp; osl_int_init(OSL_PRECISION_MP, &q_mp);
osl_int_t r_sp; osl_int_init(OSL_PRECISION_SP, &r_sp);
osl_int_t r_dp; osl_int_init(OSL_PRECISION_DP, &r_dp);
osl_int_t r_mp; osl_int_init(OSL_PRECISION_MP, &r_mp);
osl_int_floor_div_q_r(OSL_PRECISION_SP, &q_sp, &r_sp, a_sp, b_sp);
osl_int_floor_div_q_r(OSL_PRECISION_DP, &q_dp, &r_dp, a_dp, b_dp);
osl_int_floor_div_q_r(OSL_PRECISION_MP, &q_mp, &r_mp, a_mp, b_mp);
int q_sp_i = osl_int_get_si(OSL_PRECISION_SP, q_sp);
int q_dp_i = osl_int_get_si(OSL_PRECISION_DP, q_dp);
int q_mp_i = osl_int_get_si(OSL_PRECISION_MP, q_mp);
int r_sp_i = osl_int_get_si(OSL_PRECISION_SP, r_sp);
int r_dp_i = osl_int_get_si(OSL_PRECISION_DP, r_dp);
int r_mp_i = osl_int_get_si(OSL_PRECISION_MP, r_mp);
if (q_sp_i != q_dp_i || q_sp_i != q_mp_i) {
error++; printf("Error osl_int_floor_div_q_r\n");
}
if (r_sp_i != r_dp_i || r_sp_i != r_mp_i) {
error++; printf("Error osl_int_floor_div_q_r\n");
}
}
// osl_int_mod
if (!error && b_sp_i != 0) {
osl_int_mod(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_mod(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_mod(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i) {
error++; printf("Error osl_int_mod\n");
}
}
// osl_int_gcd
if (!error) {
osl_int_gcd(OSL_PRECISION_SP, &c_sp, a_sp, b_sp);
osl_int_gcd(OSL_PRECISION_DP, &c_dp, a_dp, b_dp);
osl_int_gcd(OSL_PRECISION_MP, &c_mp, a_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i) {
error++; printf("Error osl_int_gcd\n");
}
}
// osl_int_oppose
if (!error) {
osl_int_oppose(OSL_PRECISION_SP, &c_sp, b_sp);
osl_int_oppose(OSL_PRECISION_DP, &c_dp, b_dp);
osl_int_oppose(OSL_PRECISION_MP, &c_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i) {
error++; printf("Error osl_int_oppose\n");
}
}
// osl_int_abs
if (!error) {
osl_int_abs(OSL_PRECISION_SP, &c_sp, b_sp);
osl_int_abs(OSL_PRECISION_DP, &c_dp, b_dp);
osl_int_abs(OSL_PRECISION_MP, &c_mp, b_mp);
int c_sp_i = osl_int_get_si(OSL_PRECISION_SP, c_sp);
int c_dp_i = osl_int_get_si(OSL_PRECISION_DP, c_dp);
int c_mp_i = osl_int_get_si(OSL_PRECISION_MP, c_mp);
if (c_sp_i != c_dp_i || c_sp_i != c_mp_i) {
error++; printf("Error osl_int_abs\n");
}
}
// osl_int_size_in_base_2
if (!error) {
size_t r_sp = osl_int_size_in_base_2(OSL_PRECISION_SP, b_sp);
size_t r_dp = osl_int_size_in_base_2(OSL_PRECISION_DP, b_dp);
size_t r_mp = osl_int_size_in_base_2(OSL_PRECISION_MP, b_mp);
osl_int_set_si(OSL_PRECISION_SP, &c_sp, r_sp);
osl_int_set_si(OSL_PRECISION_DP, &c_dp, r_dp);
osl_int_set_si(OSL_PRECISION_MP, &c_mp, r_mp);
if (r_sp != r_dp || r_sp != r_mp) {
error++; printf("Error osl_int_size_in_base_2\n");
}
}
// osl_int_size_in_base_10
// if (!error) {
// size_t r_sp = osl_int_size_in_base_10(OSL_PRECISION_SP, b_sp);
// size_t r_dp = osl_int_size_in_base_10(OSL_PRECISION_DP, b_dp);
// size_t r_mp = osl_int_size_in_base_10(OSL_PRECISION_MP, b_mp);
//
// osl_int_set_si(OSL_PRECISION_SP, &c_sp, r_sp);
// osl_int_set_si(OSL_PRECISION_DP, &c_dp, r_dp);
// osl_int_set_si(OSL_PRECISION_MP, &c_mp, r_mp);
//
// if (r_sp != r_dp || r_sp != r_mp)
// { error++; printf("Error osl_int_size_in_base_10\n"); }
// }
// osl_int_eq
// osl_int_ne
// osl_int_pos
// osl_int_neg
// osl_int_zero
// osl_int_one
// osl_int_mone
// osl_int_divisible
if (error) {
printf("Error with:\n");
printf("\n");
printf("a_sp = "); osl_int_print(stdout, OSL_PRECISION_SP, a_sp);
printf("\n");
printf("a_dp = "); osl_int_print(stdout, OSL_PRECISION_DP, a_dp);
printf("\n");
printf("a_mp = "); osl_int_print(stdout, OSL_PRECISION_MP, a_mp);
printf("\n");
printf("\n");
printf("b_sp = "); osl_int_print(stdout, OSL_PRECISION_SP, b_sp);
printf("\n");
printf("b_dp = "); osl_int_print(stdout, OSL_PRECISION_DP, b_dp);
printf("\n");
printf("b_mp = "); osl_int_print(stdout, OSL_PRECISION_MP, b_mp);
printf("\n");
printf("\n");
printf("c_sp = "); osl_int_print(stdout, OSL_PRECISION_SP, c_sp);
printf("\n");
printf("c_dp = "); osl_int_print(stdout, OSL_PRECISION_DP, c_dp);
printf("\n");
printf("c_mp = "); osl_int_print(stdout, OSL_PRECISION_MP, c_mp);
printf("\n");
printf("\n");
nb_fail += error;
}
}
}
printf("%s ", argv[0]);
printf("fails = %d\n", nb_fail);
#else
printf("%s ", argv[0]);
printf("works only with GMP\n");
#endif
return nb_fail;
}
