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

compiler: Parametrize collapse and tiles #107

Merged
merged 3 commits into from
Jul 1, 2024

Conversation

georgebisbas
Copy link
Collaborator

No description provided.

Copy link
Collaborator

@PapyChacal PapyChacal left a comment

Choose a reason for hiding this comment

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

Yeaaaah 😋
I'm really really glad to see that part of the proper integration taking shape, thanks a lot!

Comment on lines +667 to +680
def generate_collapse_arg(num_loops: int):
"""
Generate the number of loops that will be collapsed
Resort to 1 if no number of loops is provided
"""

if num_loops < 1:
num_loops = 1

ret_arg = "collapse=" + "".join(str(num_loops)) # noqa

return ret_arg


Copy link
Collaborator

Choose a reason for hiding this comment

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

More a thought than a request 😉
It might be simpler to pass kwargs here and encapsulate the whole convert-scf-to-openmp pass? To later play with schedules, and other omp knobs like inner omp-vectorized loops and such

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

will leave this for future fixes...

@georgebisbas georgebisbas merged commit ee1326e into master Jul 1, 2024
6 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