Skip to content
Snippets Groups Projects
Commit 4ff930e3 authored by René Heß's avatar René Heß
Browse files

Make permutation and vectorization possible

parent 6023ba18
No related branches found
No related tags found
No related merge requests found
......@@ -546,25 +546,24 @@ def sum_factorization_kernel(a_matrices,
# * an input temporary (default)
# * a global data structure (if FastDGGridOperator is in use)
# * a value from a global data structure, broadcasted to a vector type (vectorized + FastDGGridOperator)
input_inames = (k_expr,) + tuple(prim.Variable(j) for j in out_inames[1:])
if l == 0 and direct_input is not None:
# See comment bellow
input_inames = _permute_backward(input_inames, perm)
globalarg(direct_input, dtype=np.float64, shape=inp_shape)
if a_matrix.vectorized:
# palpo TODO
assert(False)
input_summand = prim.Call(prim.Variable("Vec4d"),
(prim.Subscript(prim.Variable(direct_input),
(k_expr,) + tuple(prim.Variable(j) for j in out_inames[1:])),))
input_inames)))
else:
# palpo TODO
assert(False)
input_summand = prim.Subscript(prim.Variable(direct_input),
palpo + vec_iname)
input_inames + vec_iname)
else:
# If we did permute the order of a matrices above we also
# permuted the order of out_inames. Unfortunately the
# order of our input is from 0 to d-1. This means we need
# to permute _back_ to get the right coefficients.
input_inames = (k_expr,) + tuple(prim.Variable(j) for j in out_inames[1:])
if l == 0:
inp_shape = _permute_backward(inp_shape, perm)
input_inames = _permute_backward(input_inames, perm)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment