OLCF Hackathon Profiling Scripts

cpu_profile.sh

#!/bin/bash
# USAGE: Add between aprun options and executable
# For Example: aprun -n 16 -N 1 ./foo arg1 arg2
# Becomes: aprun -n 16 -N 1 ./cpu_profile.sh ./foo arg1 arg2

# Give each *rank* a separate file
LOG=cpu_profile_$ALPS_APP_PE.nvprof

# Stripe each profile file by 1 to share the load on large runs
if [ ! -f "$LOG" ] ; then
  lfs setstripe -c 1 $LOG
fi

export TMPDIR=$(mktemp -d -p $PWD)
lfs setstripe -c 1 $TMPDIR
# Execute the provided command.
exec pgprof --cpu-profiling on -o $LOG $*
rm -rf $TMPDIR

metrics.sh

#!/bin/bash
# USAGE: Add between aprun options and executable
# For Example: aprun -n 16 -N 1 ./foo arg1 arg2
# Becomes: aprun -n 16 -N 1 ./metrics.sh ./foo arg1 arg2

# Give each *rank* a separate file
LOG=metrics_$ALPS_APP_PE.nvprof

# Stripe each profile file by 1 to share the load on large runs
if [ ! -f "$LOG" ] ; then
  lfs setstripe -c 1 $LOG
fi

export TMPDIR=$(mktemp -d -p $PWD)
lfs setstripe -c 1 $TMPDIR
# Execute the provided command.
exec nvprof --analysis-metrics -o $LOG $*
rm -rf $TMPDIR

timeline.sh

#!/bin/bash
# USAGE: Add between aprun options and executable
# For Example: aprun -n 16 -N 1 ./foo arg1 arg2
# Becomes: aprun -n 16 -N 1 ./timeline.sh ./foo arg1 arg2

# Give each *rank* a separate file
LOG=timeline_$ALPS_APP_PE.nvprof

# Stripe each profile file by 1 to share the load on large runs
if [ ! -f "$LOG" ] ; then
  lfs setstripe -c 1 $LOG
fi

export TMPDIR=$(mktemp -d -p $PWD)
lfs setstripe -c 1 $TMPDIR
# Execute the provided command.
exec nvprof -o $LOG $*
rm -rf $TMPDIR