use nsys for target amip, slabplanet

CliMA · Mar 18, 2024 · cb05f3d · cb05f3d
1 parent 7014308
commit cb05f3d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git 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,9 +407,11 @@ 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"
+        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

diff --git 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"