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

Add correct cache key implementation for horizontal vectorization

parent 2172c2e6
No related branches found
No related tags found
No related merge requests found
...@@ -281,7 +281,7 @@ class SumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable): ...@@ -281,7 +281,7 @@ class SumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable):
@property @property
def function_key(self): def function_key(self):
""" Kernels sharing this key may use the same kernel implementation function """ """ Kernels sharing this key may use the same kernel implementation function """
return self.matrix_sequence return tuple(str(m) for m in self.matrix_sequence)
@property @property
def parallel_key(self): def parallel_key(self):
...@@ -544,6 +544,9 @@ class VectorizedSumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable) ...@@ -544,6 +544,9 @@ class VectorizedSumfactKernel(SumfactKernelBase, ImmutableRecord, prim.Variable)
# Some cache key definitions # Some cache key definitions
# Watch out for the documentation to see which key is used unter what circumstances # Watch out for the documentation to see which key is used unter what circumstances
# #
@property
def function_key(self):
return tuple(str(m) for m in self.matrix_sequence)
@property @property
def cache_key(self): def cache_key(self):
......
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