Skip to content

Conversation

@mwarusz
Copy link
Member

@mwarusz mwarusz commented Nov 12, 2025

This PR converts the following Omega modules to use hierarchical parallelism for limiting vertical loops extents:

  • Auxiliary state
  • Tendencies
  • EOS

It includes some minor changes to the vertical coordinate module, but #309 has more substantial updates and needs to be merged first.

Checklist

  • Documentation:
    • User's Guide has been updated
    • Developer's Guide has been updated
    • Documentation has been built locally and changes look as expected
  • Building
    • CMake build does not produce any new warnings from changes in this PR
  • Testing
    • CTest unit tests for new features have been added per the approved design.
    • Unit tests have passed. Please provide a relevant CDash build entry for verification.
    • Polaris test suite has passed

@mwarusz mwarusz changed the title Start using hierarchical parallelism to limit vertical loop extents Start using hierarchical parallelism to limit vertical loops extents Nov 12, 2025
@mwarusz
Copy link
Member Author

mwarusz commented Nov 21, 2025

I have updated this PR to use hierarchical parallelism in the time stepping code to loop only over active layers.

@brian-oneill
Copy link

Looks good as a first step to integrating hierarchical parallelism in Omega. Ran ctests successfully on Chrysalis, pm-cpu, pm-gpu, and Frontier (cpu & gpu). Thanks @mwarusz !

@mwarusz mwarusz force-pushed the omega/hipar-integration branch from 2155a08 to 726f34d Compare December 21, 2025 21:48
@mwarusz
Copy link
Member Author

mwarusz commented Dec 22, 2025

@brian-oneill

I rebased this branch and updated the docs. I had to change the mesh used in time stepper tests because of issues with forcing constant NVertLayers when the input mesh has land boundaries.

@mark-petersen mark-petersen self-assigned this Dec 22, 2025
@mark-petersen
Copy link
Collaborator

Ran a quick test. Passes CPU and GPU cTests on perlmutter.

@mark-petersen mark-petersen merged commit e3c4df5 into E3SM-Project:develop Dec 22, 2025
1 check passed
@mark-petersen
Copy link
Collaborator

Thanks @mwarusz and @brian-oneill

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