Script for gathering an nvprof timeline on a Cray XK7

nvprof_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 ./nvprof.sh ./foo arg1 arg2
export PMI_NO_FORK=1

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

# Set the process and context names
NAME="MPI Rank: %q{ALPS_APP_PE}"

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

# Execute the provided command.
exec nvprof --process-name "$NAME" --context-name "$NAME" -o "$LOG" $*