diff --git a/loopy/codegen/loop.py b/loopy/codegen/loop.py index d0d68edecf3f3c64705d8caf2afb4bf0e96697ad..3d3095d535e67161ec833549cb4b1aa0dedd1eef 100644 --- a/loopy/codegen/loop.py +++ b/loopy/codegen/loop.py @@ -401,24 +401,24 @@ def generate_sequential_loop_dim_code(codegen_state, sched_index): # }}} - # {{{ find implemented slab, build inner code + # {{{ find implemented loop, build inner code - from loopy.isl_helpers import make_loop_bound_from_pwaffs + from loopy.isl_helpers import make_loop_bounds_from_pwaffs - # impl_slab may be overapproximated - impl_slab = make_loop_bound_from_pwaffs( + # impl_loop may be overapproximated + impl_loop = make_loop_bounds_from_pwaffs( dom_and_slab.space, loop_iname, lbound, ubound) for iname in moved_inames: - dt, idx = impl_slab.get_var_dict()[iname] - impl_slab = impl_slab.move_dims( - dim_type.set, impl_slab.dim(dim_type.set), + dt, idx = impl_loop.get_var_dict()[iname] + impl_loop = impl_loop.move_dims( + dim_type.set, impl_loop.dim(dim_type.set), dt, idx, 1) new_codegen_state = ( codegen_state - .intersect(impl_slab) + .intersect(impl_loop) .copy(kernel=intersect_kernel_with_slab( kernel, slab, iname))) diff --git a/loopy/isl_helpers.py b/loopy/isl_helpers.py index 82441eb7f4b083810336916d477740693cceb012..602830de38e457c5ff4a55d7685dc346a7b4de35 100644 --- a/loopy/isl_helpers.py +++ b/loopy/isl_helpers.py @@ -102,7 +102,7 @@ def make_slab(space, iname, start, stop): return result -def make_loop_bound_from_pwaffs(space, iname, lbound, ubound): +def make_loop_bounds_from_pwaffs(space, iname, lbound, ubound): dt, pos = space.get_var_dict()[iname] iname_pwaff = isl.PwAff.var_on_domain(space, dt, pos)