Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
convolution_problem_descriptor.hpp 2.24 KB
Newer Older
Chao Liu's avatar
[MIOpen Downstream] Initial MIOpen integration (#52)  
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
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 
#ifndef CONVOLUTION_PROBLEM_DESCRIPTOR
#define CONVOLUTION_PROBLEM_DESCRIPTOR

namespace ck_driver {

struct ConvolutionProblemDescriptor
{
    ConvolutionProblemDescriptor() = default;

    ConvolutionProblemDescriptor(int N_,
                                 int K_,
                                 int C_,
                                 int Y_,
                                 int X_,
                                 int Hi_,
                                 int Wi_,
                                 int Ho_,
                                 int Wo_,
                                 int ConvStrideH_,
                                 int ConvStrideW_,
                                 int ConvDilationH_,
                                 int ConvDilationW_,
                                 int InLeftPadH_,
                                 int InLeftPadW_,
                                 int InRightPadH_,
                                 int InRightPadW_,
                                 ck::DataTypeEnum_t InDataTypeEnum_,
                                 ck::DataTypeEnum_t WeiDataTypeEnum_,
                                 ck::DataTypeEnum_t OutDataTypeEnum_)
        : N{N_},
          K{K_},
          C{C_},
          Y{Y_},
          X{X_},
          Hi{Hi_},
          Wi{Wi_},
          Ho{Ho_},
          Wo{Wo_},
          ConvStrideH{ConvStrideH_},
          ConvStrideW{ConvStrideW_},
          ConvDilationH{ConvDilationH_},
          ConvDilationW{ConvDilationW_},
          InLeftPadH{InLeftPadH_},
          InLeftPadW{InLeftPadW_},
          InRightPadH{InRightPadH_},
          InRightPadW{InRightPadW_},
          InDataTypeEnum{InDataTypeEnum_},
          WeiDataTypeEnum{WeiDataTypeEnum_},
          OutDataTypeEnum{OutDataTypeEnum_}
    {
    }

    int N;
    int K;
    int C;
    int Y;
    int X;
    int Hi;
    int Wi;
    int Ho;
    int Wo;
    int ConvStrideH;
    int ConvStrideW;
    int ConvDilationH;
    int ConvDilationW;
    int InLeftPadH;
    int InLeftPadW;
    int InRightPadH;
    int InRightPadW;

    ck::DataTypeEnum_t InDataTypeEnum;
    ck::DataTypeEnum_t WeiDataTypeEnum;
    ck::DataTypeEnum_t OutDataTypeEnum;

    std::size_t CalculateFlop() const { return 2L * N * K * C * Y * X * Ho * Wo; }
};

} // namespace ck_driver
#endif