1 /*
2 * Copyright (c) 2018-2020 Arm Limited.
3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24 #include "arm_compute/core/Types.h"
25 #include "arm_compute/core/utils/misc/ShapeCalculator.h"
26 #include "arm_compute/runtime/CL/CLTensor.h"
27 #include "arm_compute/runtime/CL/CLTensorAllocator.h"
28 #include "src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.h"
29 #include "tests/CL/CLAccessor.h"
30 #include "tests/CL/Helper.h"
31 #include "tests/PaddingCalculator.h"
32 #include "tests/datasets/ShapeDatasets.h"
33 #include "tests/framework/Asserts.h"
34 #include "tests/framework/Macros.h"
35 #include "tests/framework/datasets/Datasets.h"
36 #include "tests/validation/Validation.h"
37 #include "tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h"
38
39 namespace arm_compute
40 {
41 namespace test
42 {
43 namespace validation
44 {
45 using namespace arm_compute::misc::shape_calculator;
46
47 // Initialize the output tensor with zero and fill the border with zero
48 using CLGEMMReshapeLHSMatrix = CLSynthetizeFunctionInitOutputWithZeroAndWithZeroConstantBorder<CLGEMMReshapeLHSMatrixKernel, 16>;
49
50 template <typename T>
51 using CLGEMMReshapeLHSMatrixFixture = GEMMReshapeLHSMatrixValidationFixture<CLTensor, CLAccessor, CLGEMMReshapeLHSMatrix, T, false>;
52
53 // Fixture to use when the input has to be reinterpreted as 3D
54 template <typename T>
55 using CLGEMMReshapeLHSMatrix3DFixture = GEMMReshapeLHSMatrixValidationFixture<CLTensor, CLAccessor, CLGEMMReshapeLHSMatrix, T, true>;
56
57 // *INDENT-OFF*
58 // clang-format off
59 /** Data types */
60
61 namespace
62 {
63 /** Batch size values to test */
64 const auto b_values = framework::dataset::make("batchsize", 1, 3);
65
66 /** M0 values to test */
67 const auto m0_values_s32 = framework::dataset::make("M0", { 2, 3 });
68 const auto m0_values_s16 = framework::dataset::make("M0", { 4, 5 });
69 const auto m0_values_s8 = framework::dataset::make("M0", { 6, 7, 8 });
70
71 /** K0 values to test */
72 const auto k0_values_s32 = framework::dataset::make("K0", { 2, 3 });
73 const auto k0_values_s16 = framework::dataset::make("K0", { 4, 8 });
74 const auto k0_values_s8 = framework::dataset::make("K0", { 16 });
75
76 /** V0 values to test */
77 const auto v0_values = framework::dataset::make("V0", 1, 4);
78
79 /** Interleave values to test */
80 const auto i_values = framework::dataset::make("interleave", { true, false });
81
82 /** Transpose values to test */
83 const auto t_values = framework::dataset::make("transpose", { true, false });
84
85 } // namespace
86
87 TEST_SUITE(CL)
TEST_SUITE(GEMMReshapeLHSMatrix)88 TEST_SUITE(GEMMReshapeLHSMatrix)
89
90 FIXTURE_DATA_TEST_CASE(S32, CLGEMMReshapeLHSMatrixFixture<int>, framework::DatasetMode::ALL,
91 combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(),
92 b_values),
93 framework::dataset::make("DataType", DataType::S32)),
94 m0_values_s32),
95 k0_values_s32),
96 v0_values),
97 i_values),
98 t_values))
99 {
100 // Validate output
101 validate(CLAccessor(_target), _reference);
102 }
103 FIXTURE_DATA_TEST_CASE(S16, CLGEMMReshapeLHSMatrixFixture<short>, framework::DatasetMode::ALL,
104 combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(),
105 b_values),
106 framework::dataset::make("DataType", DataType::S16)),
107 m0_values_s16),
108 k0_values_s16),
109 v0_values),
110 i_values),
111 t_values))
112 {
113 // Validate output
114 validate(CLAccessor(_target), _reference);
115 }
116 FIXTURE_DATA_TEST_CASE(S8, CLGEMMReshapeLHSMatrixFixture<char>, framework::DatasetMode::ALL,
117 combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(),
118 b_values),
119 framework::dataset::make("DataType", DataType::S8)),
120 m0_values_s8),
121 k0_values_s8),
122 v0_values),
123 i_values),
124 t_values))
125 {
126 // Validate output
127 validate(CLAccessor(_target), _reference);
128 }
129
130 TEST_SUITE(ReinterpretInputAs3D)
131 FIXTURE_DATA_TEST_CASE(S32, CLGEMMReshapeLHSMatrix3DFixture<int>, framework::DatasetMode::ALL,
132 combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape3DShapes(),
133 b_values),
134 framework::dataset::make("DataType", DataType::S32)),
135 m0_values_s32),
136 k0_values_s32),
137 v0_values),
138 i_values),
139 t_values))
140 {
141 // Validate output
142 validate(CLAccessor(_target), _reference);
143 }
144
145 FIXTURE_DATA_TEST_CASE(S16, CLGEMMReshapeLHSMatrix3DFixture<short>, framework::DatasetMode::ALL,
146 combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape3DShapes(),
147 b_values),
148 framework::dataset::make("DataType", DataType::S16)),
149 m0_values_s16),
150 k0_values_s16),
151 v0_values),
152 i_values),
153 t_values))
154 {
155 // Validate output
156 validate(CLAccessor(_target), _reference);
157 }
158
159 FIXTURE_DATA_TEST_CASE(S8, CLGEMMReshapeLHSMatrix3DFixture<char>, framework::DatasetMode::ALL,
160 combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape3DShapes(),
161 b_values),
162 framework::dataset::make("DataType", DataType::S8)),
163 m0_values_s8),
164 k0_values_s8),
165 v0_values),
166 i_values),
167 t_values))
168 {
169 // Validate output
170 validate(CLAccessor(_target), _reference);
171 }
172 TEST_SUITE_END() // ReinterpretInputAs3D
173 TEST_SUITE_END() // GEMMReshapeLHSMatrix
174 TEST_SUITE_END() // CL
175 } // namespace validation
176 } // namespace test
177 } // namespace arm_compute
178