From 6b8e584b5f05a78765a2880df4cde74343171dbc Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 14 Sep 2015 10:17:30 -0500
Subject: [PATCH] buffer_array caching: prepare kernel for caching
---
loopy/buffer.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/loopy/buffer.py b/loopy/buffer.py
index 1e6a137b5..fea87effc 100644
--- a/loopy/buffer.py
+++ b/loopy/buffer.py
@@ -188,7 +188,9 @@ def buffer_array(kernel, var_name, buffer_inames, init_expression=None,
from loopy import CACHING_ENABLED
- cache_key = (kernel, var_name, tuple(buffer_inames),
+ from loopy.preprocess import prepare_for_caching
+ key_kernel = prepare_for_caching(kernel)
+ cache_key = (key_kernel, var_name, tuple(buffer_inames),
PymbolicExpressionHashWrapper(init_expression),
PymbolicExpressionHashWrapper(store_expression), within,
default_tag, temporary_is_local, fetch_bounding_box)
--
GitLab