conv_util.cpp 6.24 KB
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
5
6
7
8
9
10
11
12
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.

#include <iostream>
#include <string>
#include <vector>
#include <gtest/gtest.h>

#include "ck/ck.hpp"
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"

#include "ck/library/utility/check_err.hpp"
13
#include "ck/library/utility/convolution_parameter.hpp"
Chao Liu's avatar
Chao Liu committed
14
15
16
17
18
19

namespace {

class TestConvUtil : public ::testing::Test
{
    public:
Chao Liu's avatar
Chao Liu committed
20
    void SetNDParams(std::size_t ndims, std::size_t s, std::size_t d, std::size_t p)
Chao Liu's avatar
Chao Liu committed
21
    {
Chao Liu's avatar
Chao Liu committed
22
23
24
25
26
27
28
29
30
31
        conv_params = ck::utils::conv::ConvParam(ndims,
                                                 128,
                                                 192,
                                                 256,
                                                 std::vector<ck::index_t>(ndims, 3),
                                                 std::vector<ck::index_t>(ndims, 71),
                                                 std::vector<ck::index_t>(ndims, s),
                                                 std::vector<ck::index_t>(ndims, d),
                                                 std::vector<ck::index_t>(ndims, p),
                                                 std::vector<ck::index_t>(ndims, p));
Chao Liu's avatar
Chao Liu committed
32
33
34
35
36
37
    }

    protected:
    // -------  default 2D -------
    // input NCHW {128,192,71,71},
    // weights KCYX {256,192,3,3},
Chao Liu's avatar
Chao Liu committed
38
39
40
    // stride {s, s},
    // dilations {d,d},
    // padding {{p,p}, {p,p}
Chao Liu's avatar
clean  
Chao Liu committed
41
    ck::utils::conv::ConvParam conv_params;
Chao Liu's avatar
Chao Liu committed
42
43
44
45
46
47
};

} // namespace

TEST_F(TestConvUtil, ConvParamsGetOutputSpatialLengths1D)
{
Chao Liu's avatar
Chao Liu committed
48
49
    // stride 2, dilation 1, pad 1
    SetNDParams(1, 2, 1, 1);
Chao Liu's avatar
Chao Liu committed
50
51
52
53
    std::vector<ck::index_t> out_spatial_len = conv_params.GetOutputSpatialLengths();
    EXPECT_TRUE(ck::utils::check_err(
        out_spatial_len, std::vector<ck::index_t>{36}, "Error: ConvParams 1D."));

Chao Liu's avatar
Chao Liu committed
54
55
56
    // stride 1, dilation 1, pad 1
    SetNDParams(1, 1, 1, 1);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
57
58
59
    EXPECT_TRUE(ck::utils::check_err(
        out_spatial_len, std::vector<ck::index_t>{71}, "Error: ConvParams 1D stride {1}."));

Chao Liu's avatar
Chao Liu committed
60
61
62
    // stride 2, dilation 1, pad 2
    SetNDParams(1, 2, 1, 2);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
63
64
65
66
    EXPECT_TRUE(ck::utils::check_err(out_spatial_len,
                                     std::vector<ck::index_t>{37},
                                     "Error: ConvParams 1D padding left/right {2}."));

Chao Liu's avatar
Chao Liu committed
67
68
69
    // stride 2, dilation 2, pad 2
    SetNDParams(1, 2, 2, 2);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
70
71
72
    EXPECT_TRUE(ck::utils::check_err(
        out_spatial_len, std::vector<ck::index_t>{36}, "Error: ConvParams 1D dilation {2}."));

Chao Liu's avatar
Chao Liu committed
73
74
75
    // stride 3, dilation 2, pad 1
    SetNDParams(1, 3, 2, 1);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
76
77
78
79
80
81
    EXPECT_TRUE(
        ck::utils::check_err(out_spatial_len,
                             std::vector<ck::index_t>{23},
                             "Error: ConvParams 1D strides{3}, padding {1}, dilations {2}."));
}

Chao Liu's avatar
clean  
Chao Liu committed
82
83
TEST_F(TestConvUtil, ConvParamsGetOutputSpatialLengths2D)
{
Chao Liu's avatar
Chao Liu committed
84
85
    // stride 2, dilation 1, pad 1
    SetNDParams(2, 2, 1, 1);
Chao Liu's avatar
clean  
Chao Liu committed
86
87
88
89
90
    std::vector<ck::index_t> out_spatial_len = conv_params.GetOutputSpatialLengths();
    EXPECT_TRUE(ck::utils::check_err(out_spatial_len,
                                     std::vector<ck::index_t>{36, 36},
                                     "Error: ConvParams 2D default constructor."));

Chao Liu's avatar
Chao Liu committed
91
92
93
    // stride 1, dilation 1, pad 1
    SetNDParams(2, 1, 1, 1);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
clean  
Chao Liu committed
94
95
96
    EXPECT_TRUE(ck::utils::check_err(
        out_spatial_len, std::vector<ck::index_t>{71, 71}, "Error: ConvParams 2D stride {1,1}."));

Chao Liu's avatar
Chao Liu committed
97
98
99
    // stride 2, dilation 1, pad 2
    SetNDParams(2, 2, 1, 2);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
clean  
Chao Liu committed
100
101
102
103
    EXPECT_TRUE(ck::utils::check_err(out_spatial_len,
                                     std::vector<ck::index_t>{37, 37},
                                     "Error: ConvParams 2D padding left/right {2,2}."));

Chao Liu's avatar
Chao Liu committed
104
105
106
    // stride 2, dilation 2, pad 2
    SetNDParams(2, 2, 2, 2);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
clean  
Chao Liu committed
107
108
109
    EXPECT_TRUE(ck::utils::check_err(
        out_spatial_len, std::vector<ck::index_t>{36, 36}, "Error: ConvParams 2D dilation {2,2}."));

Chao Liu's avatar
Chao Liu committed
110
111
112
    // stride 3, dilation 2, pad 1
    SetNDParams(2, 3, 2, 1);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
clean  
Chao Liu committed
113
114
115
116
117
118
    EXPECT_TRUE(
        ck::utils::check_err(out_spatial_len,
                             std::vector<ck::index_t>{23, 23},
                             "Error: ConvParams 2D strides{3,3}, padding {1,1}, dilations {2,2}."));
}

Chao Liu's avatar
Chao Liu committed
119
120
TEST_F(TestConvUtil, ConvParamsGetOutputSpatialLengths3D)
{
Chao Liu's avatar
Chao Liu committed
121
122
    // stride 2, dilation 1, pad 1
    SetNDParams(3, 2, 1, 1);
Chao Liu's avatar
Chao Liu committed
123
124
125
126
    std::vector<ck::index_t> out_spatial_len = conv_params.GetOutputSpatialLengths();
    EXPECT_TRUE(ck::utils::check_err(
        out_spatial_len, std::vector<ck::index_t>{36, 36, 36}, "Error: ConvParams 3D."));

Chao Liu's avatar
Chao Liu committed
127
128
129
    // stride 1, dilation 1, pad 1
    SetNDParams(3, 1, 1, 1);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
130
131
132
133
    EXPECT_TRUE(ck::utils::check_err(out_spatial_len,
                                     std::vector<ck::index_t>{71, 71, 71},
                                     "Error: ConvParams 3D stride {1, 1, 1}."));

Chao Liu's avatar
Chao Liu committed
134
135
136
    // stride 2, dilation 1, pad 2
    SetNDParams(3, 2, 1, 2);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
137
138
139
140
    EXPECT_TRUE(ck::utils::check_err(out_spatial_len,
                                     std::vector<ck::index_t>{37, 37, 37},
                                     "Error: ConvParams 3D padding left/right {2, 2, 2}."));

Chao Liu's avatar
Chao Liu committed
141
142
143
    // stride 2, dilation 2, pad 2
    SetNDParams(3, 2, 2, 2);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
144
145
146
147
    EXPECT_TRUE(ck::utils::check_err(out_spatial_len,
                                     std::vector<ck::index_t>{36, 36, 36},
                                     "Error: ConvParams 3D dilation {2, 2, 2}."));

Chao Liu's avatar
Chao Liu committed
148
149
150
    // stride 3, dilation 2, pad 1
    SetNDParams(3, 3, 2, 1);
    out_spatial_len = conv_params.GetOutputSpatialLengths();
Chao Liu's avatar
Chao Liu committed
151
152
153
154
155
    EXPECT_TRUE(ck::utils::check_err(
        out_spatial_len,
        std::vector<ck::index_t>{23, 23, 23},
        "Error: ConvParams 3D strides{3, 3, 3}, padding {1, 1, 1}, dilations {2, 2, 2}."));
}