Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adds momentum equation test for Enzyme extension #3822

Open
wants to merge 5 commits into
base: main
Choose a base branch
from

Conversation

jlk9
Copy link
Collaborator

@jlk9 jlk9 commented Oct 4, 2024

This adds a test for Enzyme AD applied to a hydrostatic free surface model with ExplicitFreeSurface momentum advection.

Comment on lines +309 to +311
#buoyancy = buoyancy,
#tracers = (:T, :S),
#velocities = PrescribedVelocityFields(; u, v),
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
#buoyancy = buoyancy,
#tracers = (:T, :S),
#velocities = PrescribedVelocityFields(; u, v),

Copy link
Member

@glwagner glwagner left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I suggest you set up a system that allows you to loop through model properties to generate a bunch of tests automatically. For example, we can run the current test with a few advection schemes. If you set the models up with a passive tracer c, you can use derivatives of that for all configurations including for prescribed velocities and prognostic velocities.

@navidcy navidcy changed the title Added momentum equation test Added momentum equation test for Enzyme extension Oct 5, 2024
@glwagner glwagner changed the title Added momentum equation test for Enzyme extension Adds momentum equation test for Enzyme extension Oct 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants