Skip to content

[Field dimensions] Support field masking#288

Merged
gronerl merged 97 commits intoGridTools:masterfrom
jdahm:lower-dimensional-support
Jan 28, 2021
Merged

[Field dimensions] Support field masking#288
gronerl merged 97 commits intoGridTools:masterfrom
jdahm:lower-dimensional-support

Conversation

@jdahm
Copy link
Copy Markdown
Contributor

@jdahm jdahm commented Dec 15, 2020

Description

Co-authored by: @eddie-c-davis

This PR supersedes #203 and is the first in a series of PR that addresses lower dimensional storages. This PR primarily allows for lower-dimensional storages to be used in stencils, both in the annotations and in execution with storages constructed using the mask argument. This:

  • Adds frontend checks for indexing lower-dimensional fields
  • Ensures the offsets are added only for dimensions that exist in the field
  • Fixes debug, numpy, and GT backends to support lower-dimensional fields
  • Stencil object checks support storages using mask
  • Frontend and backend tests

A follow-up PR will add back the storage reduction pass.

To do:

  • Clean up the changes

jdahm and others added 30 commits September 22, 2020 17:12
…m/gt4py into bugfix/multistage-data-dependencies
Copy link
Copy Markdown
Contributor

@gronerl gronerl left a comment

Choose a reason for hiding this comment

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

In addition to the inline comments, this could need more tests, in particular also when providing arguments that are incompatible with the specification of the stencils.

Comment thread src/gt4py/frontend/gtscript_frontend.py Outdated
Comment thread src/gt4py/frontend/gtscript_frontend.py Outdated
Comment thread src/gt4py/storage/utils.py
Comment thread src/gt4py/stencil_object.py Outdated
Comment thread src/gt4py/stencil_object.py
Comment thread tests/test_integration/test_code_generation.py Outdated
@jdahm jdahm requested a review from gronerl January 27, 2021 22:42
Copy link
Copy Markdown
Contributor

@gronerl gronerl left a comment

Choose a reason for hiding this comment

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

Thanks. I also added a test for the dimension mismatch.

@gronerl gronerl merged commit 9831f6e into GridTools:master Jan 28, 2021
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.

3 participants