diff --git a/pyop2/global_kernel.py b/pyop2/global_kernel.py index 536d717e9..d49a2a5de 100644 --- a/pyop2/global_kernel.py +++ b/pyop2/global_kernel.py @@ -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 @@ -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 = {}