Skip to content

Add loop nest constraints v2 to better loop around nest map #360

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
jdsteve2 wants to merge 17 commits into
base: better_loop_around_nest_map
Choose a base branch
from
Draft

Add loop nest constraints v2 to better loop around nest map #360

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
75c1772
if iname is an IlpBaseTag then iname is also a ConcurrentTag
kaushikcfd Apr 28, 2021
e94cf78
improves the performance of find_loop_nest_around_map
kaushikcfd Apr 28, 2021
03f39a1
add treelib as a dependency
kaushikcfd Apr 28, 2021
9934dc6
convert a LoopyError to AssertionError to differentiate between user/…
kaushikcfd Apr 28, 2021
32cb282
noun -> verb for function names
kaushikcfd Apr 28, 2021
94058bb
use topological sort to get the schedule
kaushikcfd May 4, 2021
8230994
factor out generate_schedule_v2
kaushikcfd May 4, 2021
05c9ce7
only consider sequential loops for entry/exit
kaushikcfd May 4, 2021
69e6749
[schedule_v2]: be more specific in what kernels to handle, implements…
kaushikcfd May 8, 2021
77459e1
[bugfix]: fixes typos
kaushikcfd May 8, 2021
e2b41e5
import constrain_loop_nesting
jdsteve2 May 9, 2021
dbd01ca
add loop_nest_constraints attribute to kernel
jdsteve2 May 9, 2021
01c8583
copy in classes to contain loop nest constraints from branch loop-nes…
jdsteve2 May 9, 2021
f144896
copy in functions for initial loop nest constraint creation from bran…
jdsteve2 May 9, 2021
b8bde22
copy in functions for checking loop nest constraints from branch loop…
jdsteve2 May 9, 2021
0721a2c
add tests for loop nest constraint creation from branch loop-nest-con…
jdsteve2 May 9, 2021
04219ab
fix typo in comment
jdsteve2 May 9, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions loopy/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,8 @@
from loopy.version import VERSION, MOST_RECENT_LANGUAGE_VERSION

from loopy.transform.iname import (
set_loop_priority, prioritize_loops, untag_inames,
set_loop_priority, prioritize_loops, constrain_loop_nesting,
untag_inames,
split_iname, chunk_iname, join_inames, tag_inames, duplicate_inames,
rename_iname, remove_unused_inames,
split_reduction_inward, split_reduction_outward,
Expand Down Expand Up @@ -184,7 +185,8 @@

# {{{ transforms

"set_loop_priority", "prioritize_loops", "untag_inames",
"set_loop_priority", "prioritize_loops", "constrain_loop_nesting",
"untag_inames",
"split_iname", "chunk_iname", "join_inames", "tag_inames",
"duplicate_inames",
"rename_iname", "remove_unused_inames",
Expand Down
3 changes: 3 additions & 0 deletions loopy/kernel/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -272,6 +272,7 @@ def __init__(self, domains, instructions, args=None,

iname_slab_increments=None,
loop_priority=frozenset(),
loop_nest_constraints=None,
silenced_warnings=None,

applied_iname_rewrites=None,
Expand Down Expand Up @@ -417,6 +418,7 @@ def __init__(self, domains, instructions, args=None,
assumptions=assumptions,
iname_slab_increments=iname_slab_increments,
loop_priority=loop_priority,
loop_nest_constraints=loop_nest_constraints,
silenced_warnings=silenced_warnings,
temporary_variables=temporary_variables,
local_sizes=local_sizes,
Expand Down Expand Up @@ -1550,6 +1552,7 @@ def __setstate__(self, state):
"substitutions",
"iname_slab_increments",
"loop_priority",
"loop_nest_constraints",
"silenced_warnings",
"options",
"state",
Expand Down
Loading