diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml index f35cf213e8..86f8f1b925 100644 --- a/.buildkite/pipeline.yml +++ b/.buildkite/pipeline.yml @@ -380,9 +380,11 @@ steps: slurm_mem: 20GB slurm_gpus: 1 - - label: "GPU Slabplanet: albedo from static map" + - label: "GPU Slabplanet: albedo from static map (nsys)" key: "gpu_slabplanet_albedo_static_map" - command: "julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/gpu_slabplanet_albedo_static_map.yml" + command: > + nsys profile --trace=nvtx,cuda --output=experiments/AMIP/output/amip/gpu_slabplanet_albedo_static_map_artifacts + julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/gpu_slabplanet_albedo_static_map.yml artifact_paths: "experiments/AMIP/output/slabplanet/gpu_slabplanet_albedo_static_map_artifacts/*" agents: slurm_mem: 20GB @@ -405,10 +407,12 @@ steps: slurm_mem: 20GB slurm_gpus: 1 - - label: "GPU AMIP target: topography and diagnostic EDMF" + - label: "GPU AMIP target: topography and diagnostic EDMF (nsys)" key: "gpu_amip_target_topo_diagedmf_shortrun" - command: "julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/gpu_amip_target_topo_diagedmf_shortrun.yml" - artifact_paths: "experiments/AMIP/output/amip/gpu_amip_target_topo_diagedmf_shortrun_artifacts/*" + command: > + nsys profile --trace=nvtx,cuda --output=experiments/AMIP/output/amip/gpu_amip_target_topo_diagedmf_shortrun_artifacts + julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/gpu_amip_target_topo_diagedmf_shortrun.yml" + artifact_paths: "experiments/AMIP/output/amip/gpu_amip_target_topo_diagedmf_shortrun_artifacts/* agents: slurm_mem: 20GB slurm_gpus: 1 diff --git a/config/model_configs/gpu_amip_target_topo_diagedmf_shortrun.yml b/config/model_configs/gpu_amip_target_topo_diagedmf_shortrun.yml index 129e3f7409..c07a7ec759 100644 --- a/config/model_configs/gpu_amip_target_topo_diagedmf_shortrun.yml +++ b/config/model_configs/gpu_amip_target_topo_diagedmf_shortrun.yml @@ -19,7 +19,7 @@ netcdf_output_at_levels: true run_name: "gpu_amip_target_topo_diagedmf_shortrun" start_date: "19790301" surface_setup: "PrescribedSurface" -t_end: "200secs" +t_end: "12hours" topo_smoothing: true topography: "Earth" turb_flux_partition: "CombinedStateFluxes"