Skip to content
Snippets Groups Projects
Commit 42b8b04a authored by Dominic Kempf's avatar Dominic Kempf
Browse files

fixups

parent 1f00d0e1
No related branches found
No related tags found
No related merge requests found
...@@ -109,7 +109,7 @@ def pymbolic_coefficient_gradient(element, restriction, component, coeff_func, v ...@@ -109,7 +109,7 @@ def pymbolic_coefficient_gradient(element, restriction, component, coeff_func, v
for i, buf in enumerate(buffers): for i, buf in enumerate(buffers):
# Write solution from sumfactorization to gradient variable # Write solution from sumfactorization to gradient variable
assignee = prim.Subscript(prim.Variable(name), i) assignee = prim.Subscript(prim.Variable(name), i)
expression = prim.Subscript(buf, tuple(prim.Variable(i) for i in quadrature_inames())) expression = prim.Subscript(buf, vsf.quadrature_index(sf))
instruction(assignee=assignee, instruction(assignee=assignee,
expression=expression, expression=expression,
forced_iname_deps=frozenset(get_backend("quad_inames")()), forced_iname_deps=frozenset(get_backend("quad_inames")()),
...@@ -141,7 +141,7 @@ def pymbolic_coefficient(element, restriction, component, coeff_func, visitor): ...@@ -141,7 +141,7 @@ def pymbolic_coefficient(element, restriction, component, coeff_func, visitor):
var, _ = realize_sum_factorization_kernel(vsf) var, _ = realize_sum_factorization_kernel(vsf)
return prim.Subscript(var, return prim.Subscript(var,
tuple(prim.Variable(i) for i in quadrature_inames()) + vsf.vec_index_tuple(sf) vsf.quadrature_index(sf)
), visitor.indices ), visitor.indices
......
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