cubek-convolution 0.2.0

CubeK: Convolution Kernels
Documentation 
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

#[macro_export]
macro_rules! testgen_convolution_problem_size {
    ($algorithm: expr, $dtypes: expr, $tiling_scheme: expr, $swizzle: expr, $partition_buffering: expr) => {
        use $crate::suite::launcher_strategy::ConvolutionSize;

        mod g4x4x1x1 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 4,
                    w: 4,
                    c: 1,
                    out_c: 1
                }
            );
        }

        mod g17x17x1x1 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 17,
                    w: 17,
                    c: 1,
                    out_c: 1
                }
            );
        }

        mod g16x16x16x32 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 16,
                    w: 16,
                    c: 16,
                    out_c: 32
                }
            );
        }

        mod g32x32x32x16 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 32,
                    w: 32,
                    c: 32,
                    out_c: 16
                }
            );
        }

        mod g64x32x32x128 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 64,
                    w: 32,
                    c: 32,
                    out_c: 128
                }
            );
        }

        mod g32x32x64x3 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 32,
                    w: 32,
                    c: 64,
                    out_c: 3
                }
            );
        }

        mod g20x20x16x32 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 20,
                    w: 20,
                    c: 16,
                    out_c: 32
                }
            );
        }

        mod g23x10x17x20 {
            use super::*;
            $crate::testgen_convolution_launch!(
                $algorithm,
                $dtypes,
                $tiling_scheme,
                $swizzle,
                $partition_buffering,
                ConvolutionSize {
                    h: 23,
                    w: 10,
                    c: 17,
                    out_c: 20
                }
            );
        }
    };
}