add ED only, diag EDMF longruns

.buildkite/longruns/pipeline.yml

@@ -247,6 +247,62 @@ steps:
           slurm_mem: 32GB
           modules: climacommon/2025_03_18
 
+      - label: "GPU AMIP with bucket: ED only"
+        key: "amip_edonly_bucket_gpu"
+        command:
+          - echo "--- Run simulation"
+          - "julia --color=yes --project=experiments/ClimaEarth/ experiments/ClimaEarth/run_amip.jl --config_file $CONFIG_PATH/amip_edonly_bucket.yml --job_id amip_edonly_bucket_gpu"
+        artifact_paths: "experiments/ClimaEarth/output/amip_edonly_bucket_gpu/artifacts/*"
+        env:
+          CLIMACOMMS_DEVICE: "CUDA"
+        agents:
+          queue: clima
+          slurm_mem: 32GB
+          slurm_gpus: 1
+          modules: climacommon/2025_03_18
+
+      - label: "GPU AMIP with integrated land: ED only"
+        key: "amip_edonly_integrated_land_gpu"
+        command:
+          - echo "--- Run simulation"
+          - "julia --color=yes --project=experiments/ClimaEarth/ experiments/ClimaEarth/run_amip.jl --config_file $CONFIG_PATH/amip_edonly_integrated_land.yml --job_id amip_edonly_integrated_land_gpu"
+        artifact_paths: "experiments/ClimaEarth/output/amip_edonly_integrated_land_gpu/artifacts/*"
+        env:
+          CLIMACOMMS_DEVICE: "CUDA"
+        agents:
+          queue: clima
+          slurm_mem: 32GB
+          slurm_gpus: 1
+          modules: climacommon/2025_03_18
+
+      - label: "GPU AMIP with bucket: diag EDMF"
+        key: "amip_diagedmf_bucket_gpu"
+        command:
+          - echo "--- Run simulation"
+          - "julia --color=yes --project=experiments/ClimaEarth/ experiments/ClimaEarth/run_amip.jl --config_file $CONFIG_PATH/amip_diagedmf_bucket.yml --job_id amip_diagedmf_bucket_gpu"
+        artifact_paths: "experiments/ClimaEarth/output/amip_diagedmf_bucket_gpu/artifacts/*"
+        env:
+          CLIMACOMMS_DEVICE: "CUDA"
+        agents:
+          queue: clima
+          slurm_mem: 32GB
+          slurm_gpus: 1
+          modules: climacommon/2025_03_18
+
+      - label: "GPU AMIP with integrated land: diag EDMF"
+        key: "amip_diagedmf_integrated_land_gpu"
+        command:
+          - echo "--- Run simulation"
+          - "julia --color=yes --project=experiments/ClimaEarth/ experiments/ClimaEarth/run_amip.jl --config_file $CONFIG_PATH/amip_diagedmf_integrated_land.yml --job_id amip_diagedmf_integrated_land_gpu"
+        artifact_paths: "experiments/ClimaEarth/output/amip_diagedmf_integrated_land_gpu/artifacts/*"
+        env:
+          CLIMACOMMS_DEVICE: "CUDA"
+        agents:
+          queue: clima
+          slurm_mem: 32GB
+          slurm_gpus: 1
+          modules: climacommon/2025_03_18
+
   # DYAMOND AMIP: 1 day (convection resolving)
   - label: "GPU AMIP SUPERFINE: dyamond_target"
     key: "longrun_amip_dyamond_gpu"

config/longrun_configs/amip_diagedmf_bucket.yml

@@ -0,0 +1,19 @@
+FLOAT_TYPE: "Float32"
+albedo_model: "CouplerAlbedo"
+atmos_config_file: "config/longrun_configs/amip_target_diagedmf.yml"
+bucket_albedo_type: "map_temporal"
+checkpoint_dt: "720hours"
+co2: "maunaloa"
+coupler_toml: ["toml/amip.toml"]
+dt: "120secs"
+dt_cpl: "120secs"
+dt_save_to_sol: "30days"
+energy_check: false
+h_elem: 16
+mode_name: "amip"
+radiation_reset_rng_seed: true
+start_date: "20100101"
+surface_setup: "PrescribedSurface"
+t_end: "366days"
+topography: "Earth"
+topo_smoothing: true

config/longrun_configs/amip_diagedmf_integrated_land.yml

@@ -0,0 +1,19 @@
+FLOAT_TYPE: "Float32"
+albedo_model: "CouplerAlbedo"
+atmos_config_file: "config/longrun_configs/amip_target_diagedmf.yml"
+checkpoint_dt: "720hours"
+co2: "maunaloa"
+coupler_toml: ["toml/amip.toml"]
+dt: "120secs"
+dt_cpl: "120secs"
+dt_save_to_sol: "30days"
+energy_check: false
+h_elem: 16
+land_model: "integrated"
+mode_name: "amip"
+radiation_reset_rng_seed: true
+start_date: "20100101"
+surface_setup: "PrescribedSurface"
+t_end: "366days"
+topography: "Earth"
+topo_smoothing: true

config/longrun_configs/amip_edonly_bucket.yml

@@ -0,0 +1,22 @@
+FLOAT_TYPE: "Float32"
+albedo_model: "CouplerAlbedo"
+atmos_config_file: "config/longrun_configs/amip_target_edonly.yml"
+bucket_albedo_type: "map_temporal"
+checkpoint_dt: "30days"
+coupler_toml: ["toml/amip.toml"]
+dt: "120secs"
+dt_cpl: "120secs"
+dt_save_to_sol: "30days"
+energy_check: false
+h_elem: 16
+land_model: "bucket"
+mode_name: "amip"
+moist: equil
+netcdf_interpolation_num_points: [90, 45, 31]
+radiation_reset_rng_seed: true
+start_date: "20100101"
+surface_setup: "PrescribedSurface"
+t_end: "366days"
+topography: "Earth"
+topo_smoothing: true
+z_elem: 39

config/longrun_configs/amip_edonly_integrated_land.yml

@@ -0,0 +1,21 @@
+FLOAT_TYPE: "Float32"
+albedo_model: "CouplerAlbedo"
+atmos_config_file: "config/longrun_configs/amip_target_edonly.yml"
+checkpoint_dt: "30days"
+coupler_toml: ["toml/amip.toml"]
+dt: "120secs"
+dt_cpl: "120secs"
+dt_save_to_sol: "30days"
+energy_check: false
+h_elem: 16
+land_model: "integrated"
+mode_name: "amip"
+moist: equil
+netcdf_interpolation_num_points: [90, 45, 31]
+radiation_reset_rng_seed: true
+start_date: "20100101"
+surface_setup: "PrescribedSurface"
+t_end: "366days"
+topography: "Earth"
+topo_smoothing: true
+z_elem: 39