Fix microphysics issue for non-orchestrated GPU backend - 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

Fix microphysics issue for non-orchestrated GPU backend #340

Closed elynnwu closed 2 years ago

elynnwu commented 2 years ago

Purpose

This PR fixes #257.

Code changes:

microphysics.py: use cp only in orchestrated mode, regular gpu backend needs to be numpy because we cannot direct index cupy array

Infrastructure changes:

Physics dace:gpu savepoint tests are now enabled on daint, all other backends are not tested through jenkins anymore since they are tested as part of circleci.

Checklist

[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

elynnwu commented 2 years ago

launch jenkins