1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
#!/bin/bash
SOURCE="${BASH_SOURCE[0]}"
while [ -h "$SOURCE" ]; do # resolve $SOURCE until the file is no longer a symlink
DIR="$( cd -P "$( dirname "$SOURCE" )" >/dev/null && pwd )"
SOURCE="$(readlink "$SOURCE")"
[[ $SOURCE != /* ]] && SOURCE="$DIR/$SOURCE" # if $SOURCE was a relative symlink, we need to resolve it relative to the path where the symlink file was located
done
DIR="$( cd -P "$( dirname "$SOURCE" )" >/dev/null && pwd )"
if [[ -z "$BATS_LOG_DIRECTORY" ]] ; then
export BATS_LOG_DIRECTORY=$PWD
fi
if [[ ! -d $BATS_LOG_DIRECTORY ]] ; then
mkdir -p $BATS_LOG_DIRECTORY
fi
export CONTRIB_BIN=$DIR/../bin
_cancel() {
echo caught sigint
kill -s SIGKILL
}
setup() {
export BATS_TEST_LOGFILE="${BATS_LOG_DIRECTORY}/${BATS_TEST_NAME}.log"
}
teardown() {
cat ${BATS_TEST_LOGFILE} || true
}
function batch_launcher {
if [[ $# -lt 2 ]] ; then
error "not enough arguments"
fi
NUM_RANKS=$1
NUM_RANKS_PER_NODE=$2
echo "srun -n ${NUM_RANKS} --ntasks-per-node ${NUM_RANKS_PER_NODE}"
}
# necessary to avoid deadlocks during termination/interrupt
trap _cancel SIGINT
|
