Optimize column integral calculation - Githubissues

ai2cm / pace

Implementation of the FV3GFS / SHiELD atmospheric model in Python

https://ai2cm.github.io/pace/

Apache License 2.0

37 stars 12 forks source link

Optimize column integral calculation #350

Closed elynnwu closed 2 years ago

elynnwu commented 2 years ago

Purpose

This PR makes the calculation of column integrated tracers faster.

Code changes:

diagnostics: use quantity np sum instead of native python sum

Checklist

Before submitting this PR, please make sure:

[x] You have followed the coding standards guidelines established at Code Review Checklist.
[ ] Docstrings and type hints are added to new and updated routines, as appropriate
[ ] All relevant documentation has been updated or added (e.g. README, CONTRIBUTING docs)
[ ] For each public change and fix in pace-util, HISTORY has been updated
[ ] Unit tests are added or updated for non-stencil code changes