From b0b38c2f01863702764b79d9542228120fb27d09 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 26 Jan 2016 13:11:19 -0600
Subject: [PATCH] Fix buffer_array caching
---
loopy/transform/buffer.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/loopy/transform/buffer.py b/loopy/transform/buffer.py
index dd343a81a..677de78ea 100644
--- a/loopy/transform/buffer.py
+++ b/loopy/transform/buffer.py
@@ -457,13 +457,13 @@ def buffer_array(kernel, var_name, buffer_inames, init_expression=None,
from loopy import tag_inames
kernel = tag_inames(kernel, new_iname_to_tag)
+ from loopy.kernel.tools import assign_automatic_axes
+ kernel = assign_automatic_axes(kernel)
+
if CACHING_ENABLED:
from loopy.preprocess import prepare_for_caching
buffer_array_cache[cache_key] = prepare_for_caching(kernel)
- from loopy.kernel.tools import assign_automatic_axes
- kernel = assign_automatic_axes(kernel)
-
return kernel
# vim: foldmethod=marker
--
GitLab