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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
#!/usr/bin/env bash
# Author: Nico Trost
# Helper function
function display_help()
{
echo "rocSPARSE benchmark helper script"
echo " [-h|--help] prints this help message"
echo " [-d|--device] select device"
echo " [-p|--path] path to rocsparse-bench"
echo " [-m|--matrices-dir] directory of matrix files, this option discards the environment variable MATRICES_DIR. "
}
# Check if getopt command is installed
type getopt > /dev/null
if [[ $? -ne 0 ]]; then
echo "This script uses getopt to parse arguments; try installing the util-linux package";
exit 1;
fi
dev=0
path=../../build/release/clients/staging
# Parse command line parameters
getopt -T
if [[ $? -eq 4 ]]; then
GETOPT_PARSE=$(getopt --name "${0}" --longoptions help,device:,path:,matrices-dir: --options hd:p:m: -- "$@")
else
echo "Need a new version of getopt"
exit 1
fi
if [[ $? -ne 0 ]]; then
echo "getopt invocation failed; could not parse the command line";
exit 1
fi
if [[ ( ${MATRICES_DIR} == "" ) ]];then
matrices_dir=./matrices
else
matrices_dir=${MATRICES_DIR}
fi
eval set -- "${GETOPT_PARSE}"
while true; do
case "${1}" in
-m|--matrices-dir)
matrices_dir=${2}
shift 2 ;;
-h|--help)
display_help
exit 0
;;
-d|--device)
dev=${2}
shift 2 ;;
-p|--path)
path=${2}
shift 2 ;;
--) shift ; break ;;
*) echo "Unexpected command line parameter received; aborting";
exit 1
;;
esac
done
bench=$path/rocsparse-bench
# Check if binary is available
if [ ! -f $bench ]; then
echo $bench not found, exit...
exit 1
else
echo ">>" $(realpath $(ldd $bench | grep rocsparse | awk '{print $3;}'))
fi
# Generate logfile name
logname=dcsrmv_$(date +'%Y%m%d%H%M%S').log
truncate -s 0 $logname
which=`ls $matrices_dir/*.csr`
filenames=`for i in $which;do basename $i;done`
# Run csrmv for all matrices available
for filename in $filenames; do
$bench --matrices-dir $matrices_dir -f csrmv --precision d --device $dev --alpha 1 --beta 0 --iters 1000 --rocalution $filename 2>&1 | tee -a $logname
done
|
