Skip to content

Adjust gpu default block sizes in Isopycnal to account for halos and loop bounds - #87

Merged
marshallward merged 3 commits into
MOM6-GPU:dev/gpufrom
uwagura:fix/isopyncal-blocking
Sep 10, 2026
Merged

Adjust gpu default block sizes in Isopycnal to account for halos and loop bounds#87
marshallward merged 3 commits into
MOM6-GPU:dev/gpufrom
uwagura:fix/isopyncal-blocking

Conversation

@uwagura

@uwagura uwagura commented Aug 4, 2026

Copy link
Copy Markdown
Collaborator

My original implementation of blocking in MOM_isopycnal_slopes did not account for halos, or the fact that the loop bounds in this module run from js-1 to je in the meridional direction and is-1 to ie in the zonal direction. Since the default block size did not cover the entire range of these loops, we were performing multiple iterations in the zonal and meridional directions on the gpu and launching extra kernels. This PR fixes the default block sizes for gpu runs to account for the halos and loop bounds, which reduces the number of kernel launches and gives a a slight speedup.

@github-actions

github-actions Bot commented Aug 4, 2026

Copy link
Copy Markdown

GPU Port Coverage

Overall: 3344 / 7860 portable executed lines ported (42.5%)
Since base branch: +0 ported lines (+0.0 pp)

Files touched by this PR: 133 / 137 portable executed lines ported (97.1%)

Full per-file / per-routine breakdown: see the "gpu-port-report" job summary and artifact.

Full line-by-line coverage report (combined across all cases)
benchmark_ALE · double_gyre · benchmark

@JorgeG94 JorgeG94 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

looks good

Comment thread src/parameterizations/lateral/MOM_lateral_mixing_coeffs.F90 Outdated
Comment thread src/parameterizations/lateral/MOM_lateral_mixing_coeffs.F90 Outdated
@uwagura
uwagura force-pushed the fix/isopyncal-blocking branch from 4cb0ab6 to 3be7e80 Compare September 9, 2026 19:50

@marshallward marshallward left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

These changes make sense, and I detect no answer changes.

@marshallward
marshallward merged commit 75b0a16 into MOM6-GPU:dev/gpu Sep 10, 2026
53 checks passed
github-actions Bot added a commit that referenced this pull request Sep 10, 2026
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.

4 participants