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

[bugfix] Fix horizontal_index extraction

parent 0d61737c
No related branches found
No related tags found
No related merge requests found
...@@ -569,10 +569,11 @@ class VectorizedSumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable) ...@@ -569,10 +569,11 @@ class VectorizedSumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable)
return True return True
def horizontal_index(self, sf): def horizontal_index(self, sf):
key = tuple(mat.derivative for mat in sf.matrix_sequence)
for i, k in enumerate(self.kernels): for i, k in enumerate(self.kernels):
if tuple(mat.derivative for mat in k.matrix_sequence) == key: if sf.input_key == k.input_key:
return i if tuple(mat.derivative for mat in sf.matrix_sequence) == tuple(mat.derivative for mat in k.matrix_sequence):
return i
return 0 return 0
def _quadrature_index(self, sf, visitor): def _quadrature_index(self, sf, visitor):
......
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