Skip to content

merge with dev/ncar#2

Merged
alperaltuntas merged 48 commits into
alperaltuntas:introduce_cvmix_tidalfrom
NCAR:dev/ncar
Jun 1, 2018
Merged

merge with dev/ncar#2
alperaltuntas merged 48 commits into
alperaltuntas:introduce_cvmix_tidalfrom
NCAR:dev/ncar

Conversation

@alperaltuntas
Copy link
Copy Markdown
Owner

No description provided.

* I believe there is something wrong (halo updates?) with the way
this is being done. It needs to be fixed!
The background vertical diff was being added to itself, which is
wrong and was leading to a increase in kd_bkgnd over time. This
commit fixes this problem.
Total vertical viscosity can now be diagnosed at u and v points.
Variable names are now consistent with what is used in
other modules.
* Delete the old double-diffusion code
This commit adds a flag ADD_KV_SLOW (default is FALSE) that controls if
the background vertical viscosity in the interior (i.e., tidal + background
+ shear + convenction) is addded when computing the coupling coefficient.
The purpose of this flag is to be able to recover previous answers and it will
likely be removed in the future since this option should always be true.
gustavo-marques and others added 18 commits May 22, 2018 08:32
A new input parameter has been added (USE_LEGACY_DIABATIC_DRIVER). If true, the model will
use a legacy version of the diabatic driver (module MOM_legacy_diabatic_driver). This is
temporary and is needed to avoid change in answers while MOM_diabatic_driver is been
restructured.
Double diffusion cvmix + additional updates
Add old double diffusion + clean the diabatic subroutine
fix a typo in ocn_comp_mct.F90
@alperaltuntas alperaltuntas merged commit 754f4af into alperaltuntas:introduce_cvmix_tidal Jun 1, 2018
alperaltuntas pushed a commit that referenced this pull request Jul 25, 2018
use ocean instead of global pes in MOM_oda_driver
alperaltuntas pushed a commit that referenced this pull request Oct 25, 2018
Latest changes from MOM dev/gfdl
alperaltuntas pushed a commit that referenced this pull request Jan 24, 2020
alperaltuntas pushed a commit that referenced this pull request Sep 2, 2021
alperaltuntas pushed a commit that referenced this pull request Feb 27, 2026
* Add MOM_ANN module

* Mesoscale momentum parameterization with ANN

- Computes subgrid stress using ANN in MOM_Zanna_Bolton
- Uses MOM_ANN module for ANN inference

Equivalent MOM_override for defaults
```
USE_ZB2020 = True
ZB2020_USE_ANN = True
USE_CIRCULATION_IN_HORVISC = True
ZB2020_ANN_FILE_TALL = /path/to/ocean3d/subfilter/FGR3/EXP1/model/Tall.nc
```

* Mesoscale momentum parameterization with ANN (#2)

Blank commit after squash/rebase was handled on command line

* Moved MOM_ANN.F90 to src/framework/

* Minor refactor of MOM_ANN

- Removed unused modules
- Removed unused MOM_memory.h
- Added input and output means which default to 0 and
  do not need to be present in the weights file
- Gave defaults to means, norms, tests so that they do
  no need to be present in file
- Added missing array notation "(:)"
- Minor formatting

* Adds unit tests and timing test to MOM_ANN

- Added ANN_allocate, set_layer, set_input_normalization, and
  set_output_normalization methods to allow reconfiguration during
  unit tests
- Added ANN_unit_tests with some simple constructed-by-code
  networks with known solutions
- Added config_src/drivers/unit_tests/test_MOM_ANN.F90 to drive
  unit tests
- Added  config_src/drivers/timing_tests/time_MOM_ANN.F90 as
  rudimentary for timing inference

* Adding multiple forms of inference

- Adds inference operating on array (instead of single vector of
  features)
- Implements several different versions of inference with various
  loop orders
  - Involves storing the transpose of A in the type
  - Tested by checking inference on same inputs is identical between
    variants
    - Added randomizers to assist in unit testing
- Adds timing of variants to config_src/drivers/timing/time_MOM_ANN.F90
- Adds an interface (MOM_apply) to select preferred version of
  inference subroutine
- Added command line args to time_MOM_ANN.F90 to allow more rapid
  evaluation of performance

Variants explored, timed with gfortran (13.2) -O3 on Xeon:
- vector_v1:
  - original inference from Pavel
- vector_v2:
  - allocate work arrays just once, using widest layer
  - loop over layers in 2's to avoid pointer calculations and copies
  - speed up, x0.8 relative to v1
- vector_v3:
  - transpose loops
  - slow down, x1.54 relative to v1
- vector_v4:
  - transpose weights with same loop order as v1
  - slow down, x1.03 relative to v1
- array_v1:
  - same structure as v2, working on x(space,feature) input/outputs
  - speed up, x0.41 relative to v1
- array_v2:
  - as for array_v1 but with transposed loop order
  - apply activation function on vector of first index while in cache
  - speed up, x0.35 relative to v1
- array_v3:
  - same structure as v2, working on x(feature,space) input/outputs
  - speed up, x0.58 relative to v1

* Renamed ANN variants and added some module documentation

- Added module dox
- Renamed _v1, _v2 etc to labels
- Added ANN_apply_array_sio to ANN_apply interface
- Replaced "flops" with "MBps" in timing output

* Removed alternative variants of ANN in favor of optimized

- Deleted variants of ANN that did not perform as well as the two
  versions that remain.

* Apply array_sio function in ANN inference for momentum fluxes (#5)

* Apply array_sio ANN inference for computation of momentum fluxes

* remove trailing space

* Initial commit

* address Robert Hallberg code review

* Restore deafult value of ZB_SCALING coefficient

---------

Co-authored-by: Alistair Adcroft <Alistair.Adcroft@noaa.gov>
Co-authored-by: Alistair Adcroft <adcroft@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants