From 0a3096fc59bd47ff85aed96d5f0686c894494a39 Mon Sep 17 00:00:00 2001 From: kmdeck Date: Thu, 22 Aug 2024 16:19:28 -0700 Subject: [PATCH] move run to CPU --- .buildkite/pipeline.yml | 13 ++++--------- experiments/long_runs/land.jl | 2 +- src/shared_utilities/models.jl | 13 +++++++++++++ 3 files changed, 18 insertions(+), 10 deletions(-) diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml index b3656db972..08c39fb569 100644 --- a/.buildkite/pipeline.yml +++ b/.buildkite/pipeline.yml @@ -123,6 +123,10 @@ steps: agents: slurm_mem: 16G + - label: "Canopy Implicit Stepping CPU" + command: "julia --color=yes --project=.buildkite experiments/standalone/Vegetation/timestep_test.jl" + artifact_paths: "experiments/standalone/Vegetation/errors.png" + - group: "Experiments on GPU" steps: - label: "Richards Runoff GPU" @@ -134,15 +138,6 @@ steps: env: CLIMACOMMS_DEVICE: "CUDA" - - label: "Canopy Implicit Stepping GPU" - command: "julia --color=yes --project=.buildkite experiments/standalone/Vegetation/timestep_test.jl" - artifact_paths: "experiments/standalone/Vegetation/errors.png" - agents: - slurm_ntasks: 1 - slurm_gres: "gpu:1" - env: - CLIMACOMMS_DEVICE: "CUDA" - - group: "ClimaLandSimulations" steps: - label: "Ozark figures Makie" diff --git a/experiments/long_runs/land.jl b/experiments/long_runs/land.jl index db991149c0..b8a0fb814c 100644 --- a/experiments/long_runs/land.jl +++ b/experiments/long_runs/land.jl @@ -638,7 +638,7 @@ end function setup_and_solve_problem(; greet = false) t0 = 0.0 - tf = 60 * 60.0 * 24 * 60 # keep short until it runs! * 365 + tf = 60 * 60.0 * 24 * 7 # keep short until it runs! * 365 Δt = 900.0 nelements = (101, 15) if greet diff --git a/src/shared_utilities/models.jl b/src/shared_utilities/models.jl index c00ce578a2..eef7083c57 100644 --- a/src/shared_utilities/models.jl +++ b/src/shared_utilities/models.jl @@ -211,6 +211,19 @@ function make_imp_tendency(model::AbstractImExModel) return imp_tendency! end +""" + make_imp_tendency(model::AbstractModel) + +Returns an `imp_tendency` that does nothing. This model type is not +stepped explicity. +""" +function make_imp_tendency(model::AbstractModel) + compute_imp_tendency! = make_compute_imp_tendency(model) + function imp_tendency!(dY, Y, p, t) + compute_imp_tendency!(dY, Y, p, t) + end +end + """ make_exp_tendency(model::AbstractModel)