Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
pool2d_fwd.sh 2.47 KB
Newer Older
Chao Liu's avatar
GEMM+Bias+ReLU+Add (#76)  
Chao Liu 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 
#!/bin/bash

## GPU visibility
 export HIP_VISIBLE_DEVICES=0

 make -j $1

DRIVER=example/$1
VERIFY=$2
INIT=$3
REPEAT=$4

# test
########  verify  init  repeat  N__ C___ Y X Hi__ Wi__ Strides LeftPads RightPads
#$DRIVER $VERIFY $INIT $REPEAT  128  192 3 3   71   71     2 2      1 1       1 1
#$DRIVER $VERIFY $INIT $REPEAT  128   64 1 1    1    1     1 1      0 0       0 0
#$DRIVER $VERIFY $INIT $REPEAT  256    3 7 7  230  230     2 2      0 0       0 0
 $DRIVER $VERIFY $INIT $REPEAT  256 1024 14 14   14   14    1  1     0  0      0  0

 N=$5

# Resnet50
########  verify  init  repeat  N__ C___ Y X Hi__ Wi__ Strides LeftPads RightPads
#$DRIVER $VERIFY $INIT $REPEAT   $N 1024 1 1   14   14    2  2     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N 1024 1 1   14   14    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N 1024 1 1   14   14    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  128 3 3   28   28    1  1     1  1      1  1
#$DRIVER $VERIFY $INIT $REPEAT   $N  128 1 1   28   28    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  128 3 3   58   58    2  2     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N 2048 1 1    7    7    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  256 1 1   14   14    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  256 3 3   14   14    1  1     1  1      1  1
#$DRIVER $VERIFY $INIT $REPEAT   $N  256 3 3   30   30    2  2     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  256 1 1   56   56    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  256 1 1   56   56    2  2     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  256 1 1   56   56    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  512 3 3   16   16    2  2     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  512 1 1   28   28    2  2     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  512 1 1   28   28    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  512 1 1   28   28    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  512 1 1    7    7    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N  512 3 3    7    7    1  1     1  1      1  1
#$DRIVER $VERIFY $INIT $REPEAT   $N   64 1 1   56   56    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N   64 1 1   56   56    1  1     0  0      0  0
#$DRIVER $VERIFY $INIT $REPEAT   $N   64 3 3   56   56    1  1     1  1      1  1
#$DRIVER $VERIFY $INIT $REPEAT   $N    3 7 7  230  230    2  2     0  0      0  0