From 3527ef503ac621bb4b1590316257421e6f52de53 Mon Sep 17 00:00:00 2001 From: Emma Rothwell Date: Sun, 28 Jan 2024 17:02:58 +0000 Subject: [PATCH] global kernel _cache_key contains form signature --- pyop2/global_kernel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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 = {}