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

Bugfix: VectorizedSumfactKernel needs quadrature_permutation

parent 542e1317
No related branches found
No related tags found
No related merge requests found
......@@ -575,6 +575,7 @@ class VectorizedSumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable)
vertical_width=1,
buffer=None,
insn_dep=frozenset(),
quadrature_permutation=None,
):
# Assert the input data structure
assert isinstance(kernels, tuple)
......@@ -596,6 +597,11 @@ class VectorizedSumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable)
# Join the instruction dependencies of all subkernels
insn_dep = insn_dep.union(k.insn_dep for k in kernels)
# Get order of quadrature points
quadrature_permutation = kernels[0].quadrature_permutation
for k in kernels:
assert k.quadrature_permutation == quadrature_permutation
# We currently assume that all subkernels are consecutive, 0-based within the vector
assert None not in kernels
......@@ -605,6 +611,7 @@ class VectorizedSumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable)
buffer=buffer,
insn_dep=insn_dep,
vertical_width=vertical_width,
quadrature_permutation=quadrature_permutation,
)
prim.Variable.__init__(self, "VecSUMFAC")
......
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