You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
631 B
35 lines
631 B
#!/bin/bash
|
|
|
|
if [ $# -ne 4 ]; then
|
|
echo "Usage: $0 um|noum nsys|nvprof|none nodes ppn"
|
|
exit 1
|
|
fi
|
|
|
|
um=$1
|
|
prof=$2
|
|
nodes=$3
|
|
ppn=$4
|
|
|
|
tag=${um}_${prof}_${nodes}_${ppn}
|
|
|
|
if [ $prof == "nsys" ]; then
|
|
prof_cmd="nsys profile --kill=none -c cudaProfilerApi -o profile/${tag}.%q{PMIX_RANK}"
|
|
elif [ $prof == "nvprof" ]; then
|
|
prof_cmd="nvprof -o profile/nvprof.%q{PMIX_RANK}.nvvp --profile-from-start off"
|
|
else
|
|
prof_cmd=""
|
|
fi
|
|
|
|
if [ $um == "um" ]; then
|
|
cmd=./mpi_daxpy_nvtx_managed
|
|
else
|
|
cmd=./mpi_daxpy_nvtx_unmanaged
|
|
fi
|
|
|
|
total_procs=$((ppn * nodes))
|
|
|
|
set +x
|
|
mpirun -np $total_procs \
|
|
$prof_cmd $cmd >out-${tag}.txt 2>&1
|
|
set -x
|