void merge_all_types(type &l, type &r, graph &g) {
if (l.height != r.height)
std::cout << "ERROR: build_graph.cpp: merge_all_types(): unexpected types\n";
type *pl = &l;
type *pr = &r;
while (pl && pr) {
g.merge_iters(pl->dim.dim, pr->dim.dim);
pl->dim.dim = g.get_iter_rep(pl->dim.dim);
pr->dim.dim = g.get_iter_rep(pr->dim.dim);
pl = pl->t;
pr = pr->t;
}
}
void mult_sizes(type *u, type *l, type *r, graph &g) {
if (l->k == row && r->k == column) {
// operands
g.merge_iters(l->dim.dim, r->dim.dim);
r->dim.dim = g.get_iter_rep(r->dim.dim);
l->dim.dim = g.get_iter_rep(l->dim.dim);
mult_sizes(u,l->t,r->t,g);
}
else if (l->k == column && r->k == row) {
if (u->k == row) {
g.merge_iters(r->dim.dim,u->dim.dim);
r->dim.dim = g.get_iter_rep(u->dim.dim);
u->dim.dim = g.get_iter_rep(u->dim.dim);
mult_sizes(u->t, l, r->t, g);
}
else {
g.merge_iters(l->dim.dim,u->dim.dim);
l->dim.dim = g.get_iter_rep(u->dim.dim);
u->dim.dim = g.get_iter_rep(u->dim.dim);
mult_sizes(u->t, l->t, r, g);
}
}
else if ((l->k == scalar && r->k == column)
|| (l->k == scalar && r->k == row)) {
// scaling
g.merge_iters(r->dim.dim,u->dim.dim);
r->dim.dim = g.get_iter_rep(r->dim.dim);
u->dim.dim = g.get_iter_rep(u->dim.dim);
}
else if ((l->k == row && r->k == scalar)
|| (l->k == column && r->k == scalar)) {
// scaling
g.merge_iters(l->dim.dim, u->dim.dim);
l->dim.dim = g.get_iter_rep(l->dim.dim);
u->dim.dim = g.get_iter_rep(u->dim.dim);
}
else if (l->k == row && r->k == row) {
if (u->k == row) {
// result and one or other operand
g.merge_iters(u->dim.dim, r->dim.dim);
r->dim.dim = g.get_iter_rep(r->dim.dim);
u->dim.dim = g.get_iter_rep(u->dim.dim);
mult_sizes(u->t,l,r->t,g);
}
else if (u->k == column) {
// k
g.merge_iters(l->dim.dim, r->t->dim.dim);
r->t->dim.dim = g.get_iter_rep(l->dim.dim);
l->dim.dim = g.get_iter_rep(l->dim.dim);
type *tmp = new type(*(r->t));
if (tmp->k == row)
tmp->k = column;
else
tmp->k = row;
tmp->dim.dim = r->dim.dim;
tmp->dim.step = r->dim.step;
mult_sizes(u,l->t,tmp,g);
delete tmp;
}
}
else if (l->k == column && r->k == column) {
if (u->k == row) {
// k
g.merge_iters(l->t->dim.dim, r->dim.dim);
l->t->dim.dim = g.get_iter_rep(r->dim.dim);
r->dim.dim = g.get_iter_rep(r->dim.dim);
type *tmp = new type(*(l->t));
if (tmp->k == row)
tmp->k = column;
else
tmp->k = row;
tmp->dim.dim = l->dim.dim;
tmp->dim.step = l->dim.step;
mult_sizes(u,tmp,r->t,g);
delete tmp;
}
else if (u->k == column) {
// result and one or other operand
g.merge_iters(u->dim.dim, l->dim.dim);
l->dim.dim = g.get_iter_rep(u->dim.dim);
u->dim.dim = g.get_iter_rep(u->dim.dim);
mult_sizes(u->t,l->t,r,g);
}
}
else {
// must be be scalar * scalar
return;
}
}
