From 8d4a09e72b57e99b021623c41448d910805bd95e Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Wed, 5 Feb 2014 16:30:06 -0600 Subject: [PATCH] Use kernel assumptions in generating for loop bounds --- loopy/codegen/loop.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/loopy/codegen/loop.py b/loopy/codegen/loop.py index 57255c179..2004fe215 100644 --- a/loopy/codegen/loop.py +++ b/loopy/codegen/loop.py @@ -286,9 +286,16 @@ def generate_sequential_loop_dim_code(kernel, sched_index, codegen_state): # {{{ find bounds - domain = isl.align_spaces(domain, slab, across_dim_types=True, + aligned_domain = isl.align_spaces(domain, slab, across_dim_types=True, obj_bigger_ok=True) - dom_and_slab = domain & slab + + dom_and_slab = aligned_domain & slab + + assumptions_non_param = isl.BasicSet.from_params(kernel.assumptions) + dom_and_slab, assumptions_non_param = isl.align_two( + dom_and_slab, assumptions_non_param) + dom_and_slab = dom_and_slab & assumptions_non_param + del assumptions_non_param # move inames that are usable into parameters moved_inames = [] -- GitLab