File tree 1 file changed +30
-7
lines changed
1 file changed +30
-7
lines changed Original file line number Diff line number Diff line change 1
1
#! /bin/bash
2
2
3
3
gpu=V100
4
+ type=blk3d
5
+
4
6
iter=5
5
7
size=1024
6
8
indexes=$(( $size * 256 * 1024 ))
28
30
fi
29
31
if [[ " $1 " == " --pad" ]] || [[ " $2 " == " --pad" ]] || [[ " $3 " == " --pad" ]]
30
32
then
31
- pad_end=32
32
- pad_array=(0 1 2 4 8 16 32)
33
+ if [[ " $type " == " blk3d" ]]
34
+ then
35
+ pad_end=8
36
+ pad_array=(0 1 2 4 8)
37
+ else
38
+ pad_end=32
39
+ pad_array=(0 1 2 4 8 16 32)
40
+ fi
33
41
fi
34
42
if [[ " $1 " == " --halo" ]] || [[ " $2 " == " --halo" ]] || [[ " $3 " == " --halo" ]]
35
43
then
36
- halo_end=32
37
- halo_array=(0 1 2 4 8 16 32)
44
+ if [[ " $type " == " blk3d" ]]
45
+ then
46
+ halo_end=8
47
+ halo_array=(0 1 2 4 8)
48
+ else
49
+ halo_end=32
50
+ halo_array=(0 1 2 4 8 16 32)
51
+ fi
38
52
fi
39
53
40
54
echo " Type" | xargs printf " %-9s"
50
64
fi
51
65
echo
52
66
53
- type=std
54
67
ndr=1
55
68
VEC=1
56
69
if [[ $gpu == " V100" ]]
57
70
then
58
- BSIZE=512
71
+ if [[ " $type " == " blk3d" ]]
72
+ then
73
+ BSIZE=32
74
+ else
75
+ BSIZE=512
76
+ fi
59
77
gpu_id=1
60
78
max_bw=897.0
61
79
else
62
- BSIZE=128
80
+ if [[ " $type " == " blk3d" ]]
81
+ then
82
+ BSIZE=32
83
+ else
84
+ BSIZE=128
85
+ fi
63
86
gpu_id=2
64
87
max_bw=249.6
65
88
fi
You can’t perform that action at this time.
0 commit comments