petsc_project/run_benchmarks.sh at main · constracktor/petsc_project · GitHub

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
#!/bin/bash
# Set variables depending on petsc installation
export OMPI_MCA_btl_vader_single_copy_mechanism=none
export PETSC_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" >/dev/null 2>&1 && pwd )/../petsc_installations/$1/src/petsc"
export RUN_COMMAND="${PETSC_DIR}/lib/petsc/bin/petscmpiexec"
# decide for release or debug config (dependant on PETSc installation)
if [[ "$2" == "release" ]]
then
    DEBUGGING=0
    export PETSC_ARCH=arch-linux-c-opt
elif [[ "$2" == "debug" ]]
then
    DEBUGGING=1
    export PETSC_ARCH=arch-linux-c-debug
else
    echo "Please specify PETSc installation: release or debug"
    exit 1
fi
# decide to run blas benchmark or on cpu script
if [[ "$3" == "cpu" ]]
then
    CPU=1
    BLAS=0
elif [[ "$3" == "blas" ]]
then
    CPU=0
    BLAS=1
else
  echo "Please specify what to run: cpu or blas"
  exit 1
fi
if [ $BLAS == 1 ]
then
  # Compile Code
  cd src
  make petsc_blas DEBUG=${DEBUGGING}
  # Run benchmark
  cd ../benchmark_scripts
  OUTPUT_FILE_BLAS="blas_petsc.txt"
  rm $OUTPUT_FILE_BLAS
  touch $OUTPUT_FILE_BLAS
  # Run blas benchmark
  ${RUN_COMMAND} -n 1 ../src/petsc_blas | tee $OUTPUT_FILE_BLAS
elif [ $CPU == 1 ]
then
  # Compile Code
  cd src
  make petsc_cholesky DEBUG=${DEBUGGING}
  # Run script
  cd ../benchmark_scripts
  OUTPUT_FILE_CORES="cores_petsc.txt"
  OUTPUT_FILE_DATA="data_petsc.txt"
  rm $OUTPUT_FILE_CORES
  rm $OUTPUT_FILE_DATA
  #LOOP=5
  LOOP=1
  # Run cores_script from 1 to 128 cores
  START=1
  END=128
  STEP=2
  N_TRAIN=20000
  N_TEST=5000
  N_REG=100
  #./cores_script.sh $START $END $STEP $N_TRAIN $N_TEST $N_REG $LOOP $OUTPUT_FILE_CORES| tee -a $OUTPUT_FILE_CORES
  # Run data_script from 1 000 to 9 000
  START=1000
  END=9000
  STEP=1000
  N_CORES=128
  N_TEST=5000
  N_REG=100
  #./data_script.sh $START $END $STEP $N_CORES $N_TEST $N_REG $LOOP $OUTPUT_FILE_DATA| tee -a $OUTPUT_FILE_DATA
  # Run data_script from 10 000 to 100 000
  START=10000
  END=100000
  STEP=10000
  N_CORES=128
  N_TEST=5000
  N_REG=100
  #./data_script.sh $START $END $STEP $N_CORES $N_TEST $N_REG $LOOP $OUTPUT_FILE_DATA| tee -a $OUTPUT_FILE_DATA
  # Run data_script for testing
  START=10000
  END=10000
  STEP=1000
  N_CORES=8
  N_TEST=5000
  N_REG=100
  ./data_script.sh $START $END $STEP $N_CORES $N_TEST $N_REG $LOOP $OUTPUT_FILE_DATA| tee -a $OUTPUT_FILE_DATA
fi