17 lines
404 B
Bash
Executable File
17 lines
404 B
Bash
Executable File
#! /usr/bin/env bash
|
|
if [[ -n ${OMPI_COMM_WORLD_RANK+z} ]]; then
|
|
# mpich
|
|
export MPI_RANK=${OMPI_COMM_WORLD_RANK}
|
|
elif [[ -n ${MV2_COMM_WORLD_RANK+z} ]]; then
|
|
# ompi
|
|
export MPI_RANK=${MV2_COMM_WORLD_RANK}
|
|
elif [[ -n ${SLURM_PROCID+z} ]]; then
|
|
# mpich via srun
|
|
export MPI_RANK=${SLURM_PROCID}
|
|
fi
|
|
if [[ ${MPI_RANK} == "0" ]]; then
|
|
eval "omniperf profile -n testbench -- $*"
|
|
else
|
|
"$*"
|
|
fi
|