search

CliMA / ClimaAtmos.jl

ClimaAtmos.jl is a library for building atmospheric circulation models that is designed from the outset to leverage data assimilation and machine learning tools. We welcome contributions!
Apache License 2.0
72 stars 14 forks source link

O1.2.6 Demonstrate single (A100) GPU performant simulations with all parameterizations #2949

Open charleskawczynski opened 3 months ago

charleskawczynski commented 3 months ago

The Climate Modeling Alliance

Software Design Issue 📜

Purpose

Demonstrate 0.5 s per timestep at ~110 km resolution with 64 levels with all (but gravity wave) parameterizations on a single (A100) GPU.

The approach will be to tackle both egregiously expensive gpu kernels, improve parallelism where needed, and use kernel fusion to reduce sequential reads/writes.

Cost/Benefits/Risks

People and Personnel

Components

Components considered include

juliasloan25 commented 2 months ago

Is this a duplicate of https://github.com/CliMA/ClimaAtmos.jl/issues/2943 ?