Skip to content
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

Jump to bottom

Const assignment map fusion #1685

Draft
wants to merge 20 commits into
base: main
Choose a base branch
from
Draft

Const assignment map fusion #1685

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
08ffa3e
First draft: only within-state fusion, and only if the ranges exactly
pratyai Oct 8, 2024
7f9bada
add interstate fusion in there
pratyai Oct 8, 2024
94337ab
Handle the free floating maps too, whenever possible.
pratyai Oct 8, 2024
e949e8c
Handle the fact that loop variable names can be different.
pratyai Oct 9, 2024
7fd9a55
handle if the const assignments are behind an if branch that only
pratyai Oct 10, 2024
08d04fa
small changes and documentation
pratyai Oct 10, 2024
c787414
Don't use cache in tests.
pratyai Oct 10, 2024
87686ca
Fix the "no dependency" pattern + a small refactor.
pratyai Oct 10, 2024
1f40577
if the ranges are not exactly the same, still try to fuse by adding a
pratyai Oct 11, 2024
654404c
Fix the bound for grid strided loop.
pratyai Oct 11, 2024
02e8da2
Set the schedule type to the inner GSLs to sequential.
pratyai Oct 11, 2024
4e5bffe
Allow multiple nested maps, by starting to look at only the outermost…
pratyai Oct 11, 2024
58246bc
Rewrite the fusion pattern to allow fusing nested maps if possible.
pratyai Oct 11, 2024
396c84b
move out the `@staticmethod`s out of the class definiton.
pratyai Oct 11, 2024
8185fa8
Make the "grid-strided loop" a configurable option.
pratyai Oct 13, 2024
d9cc94a
Fix the range fusion (`subsets.union()` only gives a bounding box with
pratyai Oct 13, 2024
23591bd
Add checks to make sure the transformations are actually happening (or
pratyai Oct 14, 2024
73e8cd3
If there was no path established between the parent and the strided
pratyai Oct 14, 2024
3342621
Be slightly more explicit about when to use grid strided loop.
pratyai Oct 14, 2024
e1bf9a6
Handle the AST API differences between python versions.
pratyai Oct 15, 2024
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
Loading
Loading