Skip to content
Snippets Groups Projects
Commit 35981b68 authored by Matt Wala's avatar Matt Wala
Browse files

Loop bounds finder: gist against implemented domain.

parent b6c095e8
No related branches found
No related tags found
No related merge requests found
......@@ -386,17 +386,24 @@ def generate_sequential_loop_dim_code(codegen_state, sched_index):
_, loop_iname_idx = dom_and_slab.get_var_dict()[loop_iname]
impl_domain = isl.align_spaces(
codegen_state.implemented_domain,
dom_and_slab,
obj_bigger_ok=True,
across_dim_types=True
).params()
lbound = (
kernel.cache_manager.dim_min(
dom_and_slab, loop_iname_idx)
.gist(kernel.assumptions)
.gist(dom_and_slab.params())
.gist(impl_domain)
.coalesce())
ubound = (
kernel.cache_manager.dim_max(
dom_and_slab, loop_iname_idx)
.gist(kernel.assumptions)
.gist(dom_and_slab.params())
.gist(impl_domain)
.coalesce())
# }}}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment