From 11df36630f60cd3192fe4631a160429cc23fc2ac Mon Sep 17 00:00:00 2001
From: Woo-Sun Yang <wyang@lbl.gov>
Date: Sun, 28 Jul 2024 09:42:40 -0700
Subject: [PATCH] Update scripts for using Nsight Compute

---
 testcases/conus_12-km/sub_wrf_pm_testcase.sh | 2 ++
 testcases/conus_12-km/wrapper-ncu.sh         | 2 --
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/testcases/conus_12-km/sub_wrf_pm_testcase.sh b/testcases/conus_12-km/sub_wrf_pm_testcase.sh
index 80d2d6e3dd..8520d85d35 100644
--- a/testcases/conus_12-km/sub_wrf_pm_testcase.sh
+++ b/testcases/conus_12-km/sub_wrf_pm_testcase.sh
@@ -73,7 +73,9 @@ else
   (( c = (64 / (n / SLURM_JOB_NUM_NODES)) * 2 ))
   srun -n $n -c $c --cpu-bind=cores --gpus-per-task=1 --gpu-bind=none /global/common/software/m4232/pm/v4.5.2/wrf.exe
 # Profile with Nsight Compute:
+# srun --ntasks-per-node=1 dcgmi profile --pause
 # srun -n $n -c $c --cpu_bind=cores --gpus-per-task=1 --gpu-bind=none ./wrapper-ncu.sh /global/common/software/m4232/pm/v4.5.2/wrf.exe
+# srun --ntasks-per-node=1 dcgmi profile --resume
 fi
 
 #capture error code
diff --git a/testcases/conus_12-km/wrapper-ncu.sh b/testcases/conus_12-km/wrapper-ncu.sh
index daee2e3766..868dd69e1e 100755
--- a/testcases/conus_12-km/wrapper-ncu.sh
+++ b/testcases/conus_12-km/wrapper-ncu.sh
@@ -2,9 +2,7 @@
  output=report.%q{SLURM_PROCID}.%q{SLURM_JOBID}
 
  if [[ ${SLURM_PROCID} == "0" ]] ; then
-   dcgmi profile --pause
    ncu --target-processes all --kernel-id :::1 -o ${output} "$@"
-   dcgmi profile --resume
  else
    "$@"
  fi