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

GT4py.cartesian: For loop #11

Open
2 tasks
FlorianDeconinck opened this issue Mar 4, 2024 · 0 comments
Open
2 tasks

GT4py.cartesian: For loop #11

FlorianDeconinck opened this issue Mar 4, 2024 · 0 comments

Comments

@FlorianDeconinck
Copy link
Collaborator

Double for loop are a common pattern in column-physics (moist, radiation...). As of this writing the only way to emulate this behavior is to use the a while with a local variable which is convoluted and not user friendly.

Work was done at AI2 on the matter by Johann and Elynn, some of it remains in branches.

We could also limit the impact of the feature by allowing "sub-computation"on a dedicated axis e.g.

with computation(PARALLEL), interval(...)
  with sub-comptation(K, PARALLEL), interval(K, ...) or interval(..., K)
  with sub-comptation(DDIM[X], PARALLEL)

This would allow data-dimension parallelism (with restriction)


  • Implement feature in gt4py
    OR
  • Subtasks in finer grain tasks
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

No branches or pull requests

1 participant