Skip to content

Adds option to read initial temp and salt from separate files#2

Merged
alperaltuntas merged 6 commits into
NCAR:dev/gfdlfrom
gustavo-marques:temp_salt_z_init_file
Jul 6, 2017
Merged

Adds option to read initial temp and salt from separate files#2
alperaltuntas merged 6 commits into
NCAR:dev/gfdlfrom
gustavo-marques:temp_salt_z_init_file

Conversation

@gustavo-marques
Copy link
Copy Markdown
Collaborator

  • This commit adds the option to read initial temperature and salinity fields from separate z-space input files. Before, both T and S had to be in the same file. The input parameter TEMP_SALT_Z_INIT_FILE is
    kept, so previous configurations do not to be changed. Two new parameters are introduced: TEMP_Z_INIT_FILE and SALT_Z_INIT_FILE.
  • Answers should not change.

gustavo-marques and others added 6 commits June 23, 2017 13:48
Frazil formation was being converted to surface layer evaporation under ice sheves and this was double counting the effect of frazil on the system.
Removes frazil's contribution to evaporation
This commit adds the option to read initial temperature and salinity
fields from separate z-space input files. Before, both T and S had
to be in the same file. The input parameter TEMP_SALT_Z_INIT_FILE is
kept, so previous configurations do not to be changed. Two new
parameters are introduced (TEMP_Z_INIT_FILE and SALT_Z_INIT_FILE).

Answers should not change.
@alperaltuntas alperaltuntas merged commit e8f2999 into NCAR:dev/gfdl Jul 6, 2017
@alperaltuntas
Copy link
Copy Markdown
Member

Should this also be merged to dev/ncar?

@gustavo-marques
Copy link
Copy Markdown
Collaborator Author

gustavo-marques commented Jul 6, 2017 via email

@gustavo-marques
Copy link
Copy Markdown
Collaborator Author

gustavo-marques commented Jul 6, 2017 via email

alperaltuntas added a commit that referenced this pull request Jul 6, 2017
committing pull request #2 to dev/ncar branch
@alperaltuntas
Copy link
Copy Markdown
Member

done.

@gustavo-marques
Copy link
Copy Markdown
Collaborator Author

gustavo-marques commented Jul 6, 2017 via email

@gustavo-marques gustavo-marques deleted the temp_salt_z_init_file branch July 21, 2017 20:52
alperaltuntas referenced this pull request in alperaltuntas/MOM6 May 15, 2018
use ocean instead of global pes in MOM_oda_driver
gustavo-marques pushed a commit that referenced this pull request Jun 1, 2018
alperaltuntas pushed a commit that referenced this pull request Nov 7, 2018
Latest changes from MOM dev/gfdl
alperaltuntas pushed a commit that referenced this pull request Jan 7, 2020
alperaltuntas pushed a commit that referenced this pull request Apr 22, 2021
alperaltuntas pushed a commit that referenced this pull request Oct 20, 2021
alperaltuntas pushed a commit that referenced this pull request Jan 29, 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>
alperaltuntas pushed a commit that referenced this pull request Apr 2, 2026
alperaltuntas added a commit that referenced this pull request Apr 2, 2026
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