From 6d1aacf347792cc8e8f62ed1db3c0eb47e685416 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 4 Apr 2016 00:39:39 -0500
Subject: [PATCH] Assign dim_names in precompute

---
 loopy/transform/precompute.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/loopy/transform/precompute.py b/loopy/transform/precompute.py
index 45608ea8e..5c9a286de 100644
--- a/loopy/transform/precompute.py
+++ b/loopy/transform/precompute.py
@@ -783,7 +783,8 @@ def precompute(kernel, subst_use, sweep_inames=[], within=None,
                 dtype=dtype,
                 base_indices=(0,)*len(new_temp_shape),
                 shape=tuple(abm.non1_storage_shape),
-                is_local=temporary_is_local)
+                is_local=temporary_is_local,
+                dim_names=non1_storage_axis_names)
 
     else:
         temp_var = new_temporary_variables[temporary_name]
-- 
GitLab