Skip to content

Commit

Permalink
Update scripts for using Nsight Compute
Browse files Browse the repository at this point in the history
  • Loading branch information
wyang0 authored and namo626 committed Aug 11, 2024
1 parent 0fdc634 commit 9c8ce66
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 2 additions & 0 deletions testcases/conus_12-km/sub_wrf_pm_testcase.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 0 additions & 2 deletions testcases/conus_12-km/wrapper-ncu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 9c8ce66

Please sign in to comment.