RRTMGP.jl icon indicating copy to clipboard operation
RRTMGP.jl copied to clipboard

Published 20 hours ago • verified publisher icon CliMA

Improve RRTMGP.jl performance

Open simonbyrne opened this issue 1 year ago • 19 comments

TODO: set target?

simonbyrne avatar Jan 11 '24 17:01 simonbyrne

Targets for all-sky problem: RRTMGP shortwave solve time on (1) V100 for 131,658 cols at 0.98 sec RRTMGP longwave solve time on (1) V100 for 131,658 cols at 0.85 sec These targets needs to be adjusted appropriately for P100 and A100 GPUs

sriharshakandala avatar Jan 11 '24 17:01 sriharshakandala

Current status: We are currently at 1.05 sec for the shortwave solver with 131,658 cols on a A100 GPU

sriharshakandala avatar Jan 11 '24 17:01 sriharshakandala

Update OneScalar and TwoStream structs https://github.com/CliMA/RRTMGP.jl/pull/434

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Restructure Source structs https://github.com/CliMA/RRTMGP.jl/pull/436

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Simplify type parameters for AtmosphericState https://github.com/CliMA/RRTMGP.jl/pull/437

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Always inline and remove use of StepRange https://github.com/CliMA/RRTMGP.jl/pull/438

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Restructure LookUpCld to reduce metadata size https://github.com/CliMA/RRTMGP.jl/pull/442

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Ease register pressure in GasOptics.jl https://github.com/CliMA/RRTMGP.jl/pull/443

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Split shortwave optics function https://github.com/CliMA/RRTMGP.jl/pull/445

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Restructure LookUp tables https://github.com/CliMA/RRTMGP.jl/pull/449

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Add lower bound for cumulative optical thickness https://github.com/CliMA/RRTMGP.jl/pull/448

sriharshakandala avatar Feb 15 '24 15:02 sriharshakandala

Remove inferable fields from AtmosphericStates https://github.com/CliMA/RRTMGP.jl/pull/453

sriharshakandala avatar Feb 22 '24 17:02 sriharshakandala

Add CloudState https://github.com/CliMA/RRTMGP.jl/pull/454

sriharshakandala avatar Feb 22 '24 17:02 sriharshakandala

https://github.com/CliMA/RRTMGP.jl/pull/531

sriharshakandala avatar Sep 16 '24 17:09 sriharshakandala

Change data layout for fluxes https://github.com/CliMA/RRTMGP.jl/pull/532

sriharshakandala avatar Sep 16 '24 17:09 sriharshakandala

Change data layout for optical properties https://github.com/CliMA/RRTMGP.jl/pull/533

sriharshakandala avatar Sep 16 '24 17:09 sriharshakandala

Change data layout for sources https://github.com/CliMA/RRTMGP.jl/pull/534

sriharshakandala avatar Sep 16 '24 17:09 sriharshakandala

Change data layouts for pressure, temperature, col_dry and relative humidity https://github.com/CliMA/RRTMGP.jl/pull/535

sriharshakandala avatar Sep 16 '24 17:09 sriharshakandala

Change data layout for Vmr https://github.com/CliMA/RRTMGP.jl/pull/536

sriharshakandala avatar Sep 16 '24 17:09 sriharshakandala