From 0ec159e116b72a380a86877ba6116be0959bc177 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Thu, 3 Nov 2011 01:09:31 -0400 Subject: [PATCH] Make sure domain of map_space is correctly ordered in finding CSE lead index domain. --- loopy/cse.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/loopy/cse.py b/loopy/cse.py index 4fe1e9748..0bcceee64 100644 --- a/loopy/cse.py +++ b/loopy/cse.py @@ -190,8 +190,9 @@ def process_cses(kernel, lead_csed, cse_descriptors): map_space = map_space.move_dims(dim_type.in_, 0, dim_type.set, 0, ln) rn = rhs_space.dim(dim_type.set) map_space = map_space.add_dims(dim_type.out, rn) - for i, iname in enumerate(csed.independent_inames): - map_space = map_space.set_dim_name(dim_type.out, i, iname+"'") + for i in range(rhs_domain.dim(dim_type.set)): + map_space = map_space.set_dim_name(dim_type.out, i, + rhs_domain.get_dim_name(dim_type.set, i)+"'") set_space = map_space.move_dims( dim_type.out, rn, -- GitLab