run.sh 2.28 KB
Newer Older
1
2
#!/bin/bash

zjing14's avatar
zjing14 committed
3
4
5
## GPU visibility
 export ROCR_VISIBLE_DEVICE=0
 export GPU_DEVICE_ORDINAL=0
6

zjing14's avatar
zjing14 committed
7
8
## Boost
#export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
9

zjing14's avatar
zjing14 committed
10
## Compiling
11
12
13
14
15
 export OLC_DEBUG_HIP_VERBOSE=1
 export OLC_DEBUG_HIP_DUMP=1
 export OLC_DEBUG_SAVE_TEMP_DIR=1

#make -j conv_driver_v2
zjing14's avatar
zjing14 committed
16
#make -j conv_bwd_data_driver_v2
17
18
19
20
 make -j conv_driver_v2_olc

 rm -rf /root/_hip_binary_kernels_/
 rm -rf /tmp/olCompile*
21
22
23
24
25
26
27
28

LAYOUT=$1
ALGO=$2
VERIFY=$3
INIT=$4
LOG=$5
REPEAT=$6

zjing14's avatar
zjing14 committed
29
30
31
32
33
34
################################ layout  algo  verify  init  log  repeat  N__ K___ C___ Y X Hi_ Wi__ Strides Dilations LeftPads RightPads
 ./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  128  256  192 3 3  71   71     2 2       1 1      1 1       1 1
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  128  256 1024 1 7  17   17     1 1       1 1      0 3       0 3
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  256 2048 3 3  14   14     1 1       1 1      1 1       1 1
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  256  256 3 3  14   14     1 1       1 1      1 1       1 1
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  512  512 3 3   7    7     1 1       1 1      1 1       1 1
35

zjing14's avatar
zjing14 committed
36
37
38
39
40
41
42
43
44
45
46
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  128  512  192 3 3  35   35     2 2       1 1      0 0       0 0
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  256  256 3 3  30   30     2 2       1 1      0 0       0 0
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  512  512 3 3  16   16     2 2       1 1      0 0       0 0

#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256 2048 1024 1 1  14   14     2 2       1 1      0 0       0 0
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  256 1024 1 1  14   14     1 1       1 1      0 0       0 0
#./conv_driver_v2          $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  512 2048 1 1   7    7     1 1       1 1      0 0       0 0

#./conv_bwd_data_driver_v2 $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  256  256  256 3 3  14   14     1 1       1 1      1 1       1 1

 ./conv_driver_v2_olc      $LAYOUT $ALGO $VERIFY $INIT $LOG $REPEAT  128  256  192 3 3  71   71     2 2       1 1      1 1       1 1