From b061dbfa5f89396a51109b57b07189b2c1ca7ac8 Mon Sep 17 00:00:00 2001
From: Tim Warburton <timwar@caam.rice.edu>
Date: Tue, 1 Nov 2011 23:55:15 -0500
Subject: [PATCH] Don't try to adjust the storage shape of private variables.
---
loopy/preprocess.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/loopy/preprocess.py b/loopy/preprocess.py
index 2bed109f2..4d463ec3d 100644
--- a/loopy/preprocess.py
+++ b/loopy/preprocess.py
@@ -422,10 +422,15 @@ def adjust_local_temp_var_storage(kernel):
lmem_size = cl_char.usable_local_mem_size(kernel.device)
for temp_var in kernel.temporary_variables.itervalues():
+ if not temp_var.is_local:
+ new_temp_vars[temp_var.name] = temp_var.copy(storage_shape=temp_var.shape)
+ continue
+
other_loctemp_nbytes = [tv.nbytes for tv in kernel.temporary_variables.itervalues()
if tv.is_local and tv.name != temp_var.name]
storage_shape = temp_var.storage_shape
+
if storage_shape is None:
storage_shape = temp_var.shape
--
GitLab