Skip to content

Optimize ACC port of atm_compute_solve_diagnostics_work:7313#1489

Draft
abishekg7 wants to merge 3 commits into
MPAS-Dev:developfrom
abishekg7:atmosphere/opt_compute_solve_diag_7313
Draft

Optimize ACC port of atm_compute_solve_diagnostics_work:7313#1489
abishekg7 wants to merge 3 commits into
MPAS-Dev:developfrom
abishekg7:atmosphere/opt_compute_solve_diag_7313

Conversation

@abishekg7

Copy link
Copy Markdown
Collaborator

This PR introduces optimizations for the OpenACC port of atm_compute_solve_diagnostics_work:7313. The table below lists the timings for a real, global 30km experiment on A100 GPU with the nvhpc.

For nvhpc gpu runs, -gpu=math_uniform is introduced as a build flag to ensure optimizations are bit-identical, and the we report the numbers using NV_ACC_TIME=1. The GPU runs are on 1 Derecho GPU node, using 1 A100 via 1 MPI task.

The numbers reported for the CPU runs with gnu and intel compilers use the newly-added timers local to this region, and are averaged across three runs. The CPU runs use a single derecho CPU node each, fully subscribed to 128 MPI tasks.

Version GPU kernel time (ms)   CPU timer - gnu   CPU timer - intel25
base 17.534   0.007606   0.00624
1 - Add worker 6.798        
2 - swap loop seq and vector 3.656        

TODO: Need to redo/fill in CPU benchmarks add @Pranay-Reddy-Kommera as the primary author for this commit.

This work was completed in part at the NCAR/NLR/NOAA Open Hackathon, part of the Open Hackathons program. The authors would like to acknowledge OpenACC-Standard.org for their support.

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.

1 participant