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.

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

feat[next]: Limit use of global type inference in CollapseTuple pass #1355

Merged
merged 16 commits into from
Nov 6, 2023

Conversation

nfarabullini
Copy link
Contributor

@nfarabullini nfarabullini commented Nov 3, 2023

Make CollapseTuple pass configurable such that whether the global ITIR type inference is used to get the tuple size or the simple local inference can be configured using a boolean flag.

Execute with global ITIR type inference once in loop in pass manager and in all subsequent runs use the local inference. This significantly improves the performance of the toolchain.

…is used to get the tuple size or the simple heuristics can be configured using a boolean flag to the pass

Execute with ITIR type inference once in loop in pass manager and in all subsequent runs use the simple heuristics
@nfarabullini nfarabullini force-pushed the revert_tuple_get_edits branch from 0af8960 to 555a40d Compare November 3, 2023 11:45
@nfarabullini nfarabullini changed the title reverted some edits for collapse_tuple collapse_tuple configurable for ITIR type inference and simple heuristics Nov 3, 2023
Copy link
Contributor

@tehrengruber tehrengruber left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Only style suggestions.

src/gt4py/next/iterator/transforms/collapse_tuple.py Outdated Show resolved Hide resolved
src/gt4py/next/iterator/transforms/collapse_tuple.py Outdated Show resolved Hide resolved
src/gt4py/next/iterator/transforms/collapse_tuple.py Outdated Show resolved Hide resolved
src/gt4py/next/iterator/transforms/collapse_tuple.py Outdated Show resolved Hide resolved
src/gt4py/next/iterator/transforms/collapse_tuple.py Outdated Show resolved Hide resolved
src/gt4py/next/iterator/transforms/pass_manager.py Outdated Show resolved Hide resolved
@tehrengruber tehrengruber changed the title collapse_tuple configurable for ITIR type inference and simple heuristics feat[next]: Limit use of global type inference in CollapseTuple pass Nov 5, 2023
@nfarabullini nfarabullini merged commit 4d8df69 into GridTools:main Nov 6, 2023
18 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants