O1.2.1 (part of) GPU support for CloudMicrophysics.jl

[trontrytel]

Objective

Make all CloudMicrophysics.jl functions GPU compatible.

People and Personnel

• Lead: @sriharshakandala
• Collaborators: @trontrytel
• Reviewers: @simonbyrne

Components

• CloudMicrophysics.jl already has tests that run on the GPU. The majority of functions are simple point-wise functions that should not pose any issues.
• Might benefit from refactoring how we handle the parameter struct.
• The only thing we really need to decide right now is how to handle asserts and warnings.

Inputs

• Decide how to handle asserts and warnings for when someone is calling a parameterization outside of it's accepted range of inputs.

Results and Deliverables

• Do we want to have above 90% GPU test code coverage? Or is that an overkill, given that we have a very high CPU test coverage? Is it enough for the GPU tests to test an example function from each module?

Task Breakdown And Schedule

Going through the modules in the order of how we will need them:

• Break up parameters struct in CloudMicrophysics.jl 0, 1, 2 - moment microphysics schems #160
• GPU check Microphysics0M #199
• GPU check Microphysics1M #219
• Add GPU tests for CloudMicrophysics.jl 2-moment scheme #153
• Add a custom gamma function (?) #242
• GPU check Microphysics2M #233
• Make P3 scheme GPU compatible #177

Additional items (would be nice, but not strictly necessary)

• Update CloudMicrophysics.jl GPU tests #173
• Logging variable values outside of valid ranges in GPU kernels #234

SDI Revision Log

CC

@tapios @simonbyrne @cmbengue

[trontrytel]

Checking this as done.

Tackling the logging will take longer and it's not immediately needed to run on the GPU