Skip to content

Commit

Permalink
global kernel _cache_key contains form signature
Browse files Browse the repository at this point in the history
  • Loading branch information
emmarothwell1 committed Jan 28, 2024
1 parent 26f2e51 commit 3527ef5
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion pyop2/global_kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -293,7 +293,8 @@ def __init__(self, local_kernel, arguments, *,
constant_layers=False,
subset=False,
iteration_region=None,
pass_layer_arg=False):
pass_layer_arg=False,
form_signature=None):
if self._initialized:
return

Expand Down Expand Up @@ -328,6 +329,7 @@ def __init__(self, local_kernel, arguments, *,
self._subset = subset
self._iteration_region = iteration_region
self._pass_layer_arg = pass_layer_arg
self._form_signature = form_signature

# Cache for stashing the compiled code
self._func_cache = {}
Expand Down

0 comments on commit 3527ef5

Please sign in to comment.