diff --git a/loopy/codegen/loop.py b/loopy/codegen/loop.py index ad80475c1d27f67b3df8a885f60dd96ff28efe6a..2716730b92ce4da6b43d186598dd3b577ac45282 100644 --- a/loopy/codegen/loop.py +++ b/loopy/codegen/loop.py @@ -376,10 +376,10 @@ def generate_sequential_loop_dim_code(codegen_state, sched_index): # move inames that are usable into parameters moved_inames = [] - for iname in dom_and_slab.get_var_names(dim_type.set): - if iname in usable_inames: - moved_inames.append(iname) - dt, idx = dom_and_slab.get_var_dict()[iname] + for das_iname in dom_and_slab.get_var_names(dim_type.set): + if das_iname in usable_inames: + moved_inames.append(das_iname) + dt, idx = dom_and_slab.get_var_dict()[das_iname] dom_and_slab = dom_and_slab.move_dims( dim_type.param, dom_and_slab.dim(dim_type.param), dt, idx, 1) @@ -422,8 +422,8 @@ def generate_sequential_loop_dim_code(codegen_state, sched_index): impl_lbound, impl_ubound) - for iname in moved_inames: - dt, idx = impl_loop.get_var_dict()[iname] + for moved_iname in moved_inames: + dt, idx = impl_loop.get_var_dict()[moved_iname] impl_loop = impl_loop.move_dims( dim_type.set, impl_loop.dim(dim_type.set), dt, idx, 1) @@ -432,7 +432,7 @@ def generate_sequential_loop_dim_code(codegen_state, sched_index): codegen_state .intersect(impl_loop) .copy(kernel=intersect_kernel_with_slab( - kernel, slab, iname))) + kernel, slab, loop_iname))) inner = build_loop_nest(new_codegen_state, sched_index+1)