Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
test_convnd_fwd.sh 2.65 KB
Newer Older
Adam Osewski's avatar
Unified conv3D API + support for all data types. (#133)  
Adam Osewski committed
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 
#!/usr/bin/env bash

# set -e

DIM1=False
DIM2=True
DIM3=False
DATE=220317
GIT_HASH=4e6dfda
LOG_DIR=${DATE}_${GIT_HASH}
SUFFIX=${GIT_HASH}


#--------------------------------------------------------------------------
#   Commandline arguments parsing
#   like: cmd -key[--key] value
#--------------------------------------------------------------------------

POSITIONAL=()
while [[ $# -gt 0 ]]
do
key="$1"

case $key in
    -d1|--d1)
    DIM1=True
    echo DIM1: "${DIM1}"
    shift # past argument
    ;;
    -d2|--d2)
    DIM2=True
    echo DIM2: "${DIM2}"
    shift # past argument
    ;;
    -d3|--d3)
    DIM3=True
    echo DIM3: "${DIM3}"
    shift # past argument
    ;;
    -all|--all)
    DIM1=True
    DIM2=True
    DIM3=True
    echo DIM1: "${DIM1}"
    echo DIM2: "${DIM2}"
    echo DIM3: "${DIM3}"
    shift # past argument
    ;;
    -s|--suffix)
    SUFFIX=${SUFFIX}_"$2"
    echo SUFFIX: "${SUFFIX}"
    shift # past argument
    shift # past value
    ;;
    *)    # unknown option
    POSITIONAL+=("$1") # save it in an array for later
    shift # past argument
    ;;
esac
done
set -- "${POSITIONAL[@]}" # restore positional parameters

#--------------------------------------------------------------------------

# NUMACTL="numactl --cpunodebind=1 --membind=1"
NUMACTL=
# ENV_CONF=
Illia Silin's avatar
Code clean-up (#1285)  
Illia Silin committed
68 
GPU=gfx908
Adam Osewski's avatar
Unified conv3D API + support for all data types. (#133)  
Adam Osewski committed
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110 
PROF_ITER_COUNT=10000
LOG_DIR_PATH=../log/${LOG_DIR}
set -x

#-------------------------------------------------------------------------------
#               1D
#-------------------------------------------------------------------------------

if [[ "${DIM1}" == "True" ]]; then
    mkdir -p ${LOG_DIR_PATH}
    echo ">>>>>>>> RUN test conv1d nwc <<<<<<<<<<"
    CMD="./../build/bin/test_conv1d_fwd"
    ${NUMACTL} ${CMD} 2>&1 \
        | tee ${LOG_DIR_PATH}/test_conv1d_fwd_nwc_${SUFFIX}_${GPU}.log

fi

#-------------------------------------------------------------------------------
#               2D
#-------------------------------------------------------------------------------

if [[ "${DIM2}" == "True" ]]; then
    mkdir -p ${LOG_DIR_PATH}
    echo ">>>>>>>> RUN test conv2d nhwc <<<<<<<<<<"
    CMD="./../build/bin/test_conv2d_fwd"
    ${NUMACTL} ${CMD} 2>&1 \
        | tee ${LOG_DIR_PATH}/test_conv2d_fwd_nhwc_${SUFFIX}_${GPU}.log

fi

#-------------------------------------------------------------------------------
#               3D
#-------------------------------------------------------------------------------

if [[ "${DIM3}" == "True" ]]; then
    mkdir -p ${LOG_DIR_PATH}
    echo ">>>>>>>> RUN test conv3d ndhwc <<<<<<<<<<"
    CMD="./../build/bin/test_conv3d_fwd"
    ${NUMACTL} ${CMD} 2>&1 \
        | tee ${LOG_DIR_PATH}/test_conv3d_fwd_ndhwc_${SUFFIX}_${GPU}.log

fi