From d90311c3a30967e457706204d246adbab2f764cd Mon Sep 17 00:00:00 2001 From: JDBetteridge Date: Mon, 22 Jun 2026 18:53:03 +0000 Subject: [PATCH] compiler: Add buffer to kwargs --- devito/passes/clusters/buffering.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/devito/passes/clusters/buffering.py b/devito/passes/clusters/buffering.py index 02fe2144f3..76ce0ff610 100644 --- a/devito/passes/clusters/buffering.py +++ b/devito/passes/clusters/buffering.py @@ -1037,7 +1037,8 @@ def _select_buffer(f, dim, guard, cgroup, xds, async_degree, sregistry, callback assert len(buffers) == 1, "Unexpected form of multi-level buffering" buffer, = buffers xd = buffer.indices[dim] - extra_kwargs = {'is_autopaddable': buffer.is_autopaddable} + extra_kwargs = {'is_autopaddable': buffer.is_autopaddable, + 'buffer': buffer} else: size = infer_buffer_size(f, dim, cgroup) if async_degree is not None: