Gitiles
Code Review Sign In
review.mlplatform.org / ml / ComputeLibrary / b238f5f720be59c10b6caa633a481d0d9a3cc7a0 / . / tests / validation / CL / GEMMMatrixMultiplyReshaped.cpp
blob: 25221451ed47571233eac10a720d14447da784d3 [file] [log] [blame]
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]1/*
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]2 * Copyright (c) 2018-2019 ARM Limited.
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]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/CL/kernels/CLGEMMMatrixMultiplyReshapedKernel.h"
25#include "arm_compute/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.h"
26#include "arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h"
Gian Marco Iodice7026b302019-06-26 17:18:11 +0100[diff] [blame]27#include "arm_compute/core/KernelDescriptors.h"
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]28#include "arm_compute/core/Types.h"
29#include "arm_compute/core/utils/misc/ShapeCalculator.h"
30#include "arm_compute/runtime/CL/CLTensor.h"
31#include "arm_compute/runtime/CL/CLTensorAllocator.h"
32#include "tests/CL/CLAccessor.h"
33#include "tests/CL/Helper.h"
34#include "tests/PaddingCalculator.h"
35#include "tests/datasets/ShapeDatasets.h"
36#include "tests/framework/Asserts.h"
37#include "tests/framework/Macros.h"
38#include "tests/framework/datasets/Datasets.h"
39#include "tests/validation/Validation.h"
40#include "tests/validation/fixtures/GEMMFixture.h"
41
42namespace arm_compute
43{
44namespace test
45{
46namespace validation
47{
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]48using namespace arm_compute::misc::shape_calculator;
49
50// Create function for CLGEMMReshapeLHSMatrixKernel
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]51using CLGEMMReshapeLHSMatrix = CLSynthetizeFunction<CLGEMMReshapeLHSMatrixKernel>;
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]52
53// Create function for CLGEMMReshapeRHSMatrixKernel
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]54using CLGEMMReshapeRHSMatrix = CLSynthetizeFunction<CLGEMMReshapeRHSMatrixKernel>;
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]55
56// Create function for CLGEMMMatrixMultiplyReshapedKernel
57using CLGEMMMatrixMultiplyReshaped = CLSynthetizeFunction<CLGEMMMatrixMultiplyReshapedKernel>;
58
59// Fixture for CLGEMMMatrixMultiplyReshaped
60template <typename T>
61using CLGEMMMatrixMultiplyReshapedFixture = GEMMMatrixMultiplyReshapedValidationFixture<CLTensor, CLAccessor, T, CLGEMMReshapeLHSMatrix, CLGEMMReshapeRHSMatrix, CLGEMMMatrixMultiplyReshaped>;
62
63// Fixture for CLGEMMMatrixMultiplyReshaped3D
64template <typename T>
65using CLGEMMMatrixMultiplyReshaped3DFixture = GEMMMatrixMultiplyReshaped3DValidationFixture<CLTensor, CLAccessor, T, CLGEMMReshapeLHSMatrix, CLGEMMReshapeRHSMatrix, CLGEMMMatrixMultiplyReshaped>;
66
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]67namespace
68{
69// *INDENT-OFF*
70// clang-format off
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]71RelativeTolerance<float> rel_tolerance_f32(0.001f);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]72constexpr float abs_tolerance_f32(0.0001f);
73
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]74RelativeTolerance<half> rel_tolerance_f16(half(0.2));
75constexpr float tolerance_num_f16 = 0.02f;
76
77/** Alpha values to test - Precommit */
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]78const auto a_values = framework::dataset::make("alpha", {1.0f, -0.75f} );
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]79
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]80/** Beta values to test - Precommit */
81const auto beta_values = framework::dataset::make("beta", {-0.75f, 0.0f} );
82
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]83/** M values to test */
84const auto m_values = framework::dataset::make("M", 37);
85
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]86/** M_W values to test */
87const auto m_w_values = framework::dataset::make("M_W", 5);
88
89/** M_H values to test */
90const auto m_h_values = framework::dataset::make("M_H", 7);
91
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]92/** N values to test */
93const auto n_values = framework::dataset::make("N", 51);
94
95/** K values to test */
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]96const auto k_values = framework::dataset::make("K", 23);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]97
98/** Batch size values to test */
99const auto b_values = framework::dataset::make("batch_size", 1, 3);
100
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]101/** Activation values to test */
102const auto act_values = framework::dataset::make("Activation",
103{
104 ActivationLayerInfo(),
105 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::LU_BOUNDED_RELU, 8.f, 2.f),
106});
107
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]108/** M0 values to test - Precommit */
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]109const auto m0_values_precommit = framework::dataset::make("M0", {4, 6});
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]110
111/** N0 values to test - Precommit */
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]112const auto n0_values_precommit = framework::dataset::make("N0", { 2, 4 });
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]113
114/** K0 values to test - Precommit */
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]115const auto k0_values_precommit = framework::dataset::make("K0", { 4 });
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]116
117/** V0 values to test - Precommit */
118const auto v0_values_precommit = framework::dataset::make("V0", 1, 3);
119
120/** H0 values to test - Precommit */
121const auto h0_values_precommit = framework::dataset::make("H0", 1, 3);
122
123/** M0 values to test - Nightly */
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]124const auto m0_values_nightly = framework::dataset::make("M0", 2, 7);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]125
126/** N0 values to test - Nightly */
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]127const auto n0_values_nightly = framework::dataset::make("N0", { 2, 3, 4, 8 });
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]128
129/** K0 values to test - Nightly */
Gian Marco Iodicebacfec52019-01-11 11:30:55 +0000[diff] [blame]130const auto k0_values_nightly = framework::dataset::make("K0", { 2, 3, 4, 8 });
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]131
132/** V0 values to test - Nightly */
133const auto v0_values_nightly = framework::dataset::make("V0", 1, 4);
134
135/** H0 values to test - Nightly */
136const auto h0_values_nightly = framework::dataset::make("H0", 1, 4);
137
138/** Interleave values to test with LHS matrix */
139const auto i_values_lhs = framework::dataset::make("interleave_lhs", { true, false });
140
141/** Interleave values to test with RHS matrix */
142const auto i_values_rhs = framework::dataset::make("interleave_rhs", { true, false });
143
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]144/** Broadcast bias from vector to matrix */
145const auto broadcast_bias_values = framework::dataset::make("broadcast_bias", {false, true} );
146
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]147/** Configuration test */
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]148void validate_configuration(unsigned int m_value, unsigned int n_value, unsigned int k_value, unsigned int b_value, unsigned int m0_value, unsigned int n0_value, unsigned int k0_value, unsigned int v0_value, unsigned int h0_value, bool i_value_lhs, bool i_value_rhs, bool broadcast_bias, DataType data_type, const ActivationLayerInfo &act_info)
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]149{
150 const unsigned int M = m_value;
151 const unsigned int N = n_value;
152 const unsigned int K = k_value;
153
154 GEMMLHSMatrixInfo lhs_info;
155 lhs_info.m0 = m0_value;
156 lhs_info.k0 = k0_value;
157 lhs_info.v0 = v0_value;
158 lhs_info.interleave = i_value_lhs;
159 lhs_info.transpose = false;
160
161 GEMMRHSMatrixInfo rhs_info;
162 rhs_info.n0 = n0_value;
163 rhs_info.k0 = k0_value;
164 rhs_info.h0 = h0_value;
165 rhs_info.interleave = i_value_rhs;
166 rhs_info.transpose = true;
167
Gian Marco Iodice7026b302019-06-26 17:18:11 +0100[diff] [blame]168 GEMMKernelInfo kernel_info;
169 kernel_info.m = M;
170 kernel_info.n = N;
171 kernel_info.k = K;
172 kernel_info.depth_output_gemm3d = 0;
173 kernel_info.reinterpret_input_as_3d = false;
174 kernel_info.broadcast_bias = broadcast_bias;
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]175 kernel_info.activation_info = act_info;
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]176
177 const TensorShape lhs_shape(K, M, b_value);
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]178 const TensorShape lhs_shape_reshaped = compute_lhs_reshaped_shape(TensorInfo(lhs_shape, 1, data_type),
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]179 lhs_info,
180 false);
181
182 const TensorShape rhs_shape(N, K, b_value);
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]183 const TensorShape rhs_shape_reshaped = compute_rhs_reshaped_shape(TensorInfo(rhs_shape, 1, data_type),
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]184 rhs_info);
185
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]186 const TensorShape dst_shape = compute_mm_shape(TensorInfo(lhs_shape_reshaped, 1, data_type),
187 TensorInfo(rhs_shape_reshaped, 1, data_type),
Gian Marco Iodice7026b302019-06-26 17:18:11 +0100[diff] [blame]188 kernel_info);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]189
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]190 const TensorShape bias_shape(N,
191 broadcast_bias? 1 : M,
192 broadcast_bias? 1 : b_value);
193
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]194 // Create tensors
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]195 CLTensor lhs_reshaped = create_tensor<CLTensor>(lhs_shape_reshaped, data_type);
196 CLTensor rhs_reshaped = create_tensor<CLTensor>(rhs_shape_reshaped, data_type);
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]197 CLTensor bias = create_tensor<CLTensor>(bias_shape, data_type);
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]198 CLTensor dst = create_tensor<CLTensor>(dst_shape, data_type);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]199
200 ARM_COMPUTE_EXPECT(lhs_reshaped.info()->is_resizable(), framework::LogLevel::ERRORS);
201 ARM_COMPUTE_EXPECT(rhs_reshaped.info()->is_resizable(), framework::LogLevel::ERRORS);
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]202 ARM_COMPUTE_EXPECT(bias.info()->is_resizable(), framework::LogLevel::ERRORS);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]203 ARM_COMPUTE_EXPECT(dst.info()->is_resizable(), framework::LogLevel::ERRORS);
204
205 // Create and configure function
206 CLGEMMMatrixMultiplyReshaped gemm;
Gian Marco Iodice7026b302019-06-26 17:18:11 +0100[diff] [blame]207 gemm.configure(&lhs_reshaped, &rhs_reshaped, &bias, &dst, 1.0f, 1.0f, lhs_info, rhs_info, kernel_info);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]208}
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]209} // namespace
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]210
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]211TEST_SUITE(CL)
212TEST_SUITE(GEMMMatrixMultiplyReshaped)
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]213TEST_SUITE(Float)
214TEST_SUITE(FP32)
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]215DATA_TEST_CASE(Configuration, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]216 m_values,
217 n_values),
218 k_values),
219 framework::dataset::make("batch_size", 1)),
220 m0_values_precommit),
221 n0_values_precommit),
222 k0_values_precommit),
223 v0_values_precommit),
224 h0_values_precommit),
225 i_values_lhs),
226 i_values_rhs),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]227 broadcast_bias_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]228 act_values),
229m_value, n_value, k_value, b_value, m0_value, n0_value, k0_value, v0_value, h0_value, i_value_lhs, i_value_rhs, broadcast_bias, act_value)
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]230{
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]231 validate_configuration(m_value, n_value, k_value, b_value, m0_value, n0_value, k0_value, v0_value, h0_value, i_value_lhs, i_value_rhs, broadcast_bias, DataType::F32, act_value);
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]232}
233
234FIXTURE_DATA_TEST_CASE(RunSmall, CLGEMMMatrixMultiplyReshapedFixture<float>, framework::DatasetMode::ALL,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]235 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]236 m_values,
237 n_values),
238 k_values),
239 b_values),
240 m0_values_precommit),
241 n0_values_precommit),
242 k0_values_precommit),
243 v0_values_precommit),
244 h0_values_precommit),
245 i_values_lhs),
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]246 i_values_rhs),
247 framework::dataset::make("DataType", DataType::F32)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]248 a_values),
249 beta_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]250 broadcast_bias_values),
251 act_values))
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]252{
253 // Validate output
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]254 validate(CLAccessor(_target), _reference, rel_tolerance_f32, 0.f, abs_tolerance_f32);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]255}
256
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]257FIXTURE_DATA_TEST_CASE(RunLarge, CLGEMMMatrixMultiplyReshapedFixture<float>, framework::DatasetMode::NIGHTLY,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]258 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]259 m_values,
260 n_values),
261 k_values),
262 b_values),
263 m0_values_nightly),
264 n0_values_nightly),
265 k0_values_nightly),
266 v0_values_nightly),
267 h0_values_nightly),
268 i_values_lhs),
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]269 i_values_rhs),
270 framework::dataset::make("DataType", DataType::F32)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]271 a_values),
272 beta_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]273 broadcast_bias_values),
274 act_values))
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]275{
276 // Validate output
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]277 validate(CLAccessor(_target), _reference, rel_tolerance_f32, 0.f, abs_tolerance_f32);
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]278}
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]279
280FIXTURE_DATA_TEST_CASE(RunSmall3D, CLGEMMMatrixMultiplyReshaped3DFixture<float>, framework::DatasetMode::ALL,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]281 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]282 m_w_values,
283 m_h_values),
284 n_values),
285 k_values),
286 b_values),
287 m0_values_precommit),
288 n0_values_precommit),
289 k0_values_precommit),
290 v0_values_precommit),
291 h0_values_precommit),
292 i_values_lhs),
293 i_values_rhs),
294 framework::dataset::make("DataType", DataType::F32)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]295 a_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]296 beta_values),
297 act_values))
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]298{
299 // Validate output
300 validate(CLAccessor(_target), _reference, rel_tolerance_f32, 0.f, abs_tolerance_f32);
301}
302
303FIXTURE_DATA_TEST_CASE(RunLarge3D, CLGEMMMatrixMultiplyReshaped3DFixture<float>, framework::DatasetMode::NIGHTLY,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]304 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]305 m_w_values,
306 m_h_values),
307 n_values),
308 k_values),
309 b_values),
310 m0_values_nightly),
311 n0_values_nightly),
312 k0_values_nightly),
313 v0_values_nightly),
314 h0_values_nightly),
315 i_values_lhs),
316 i_values_rhs),
317 framework::dataset::make("DataType", DataType::F32)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]318 a_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]319 beta_values),
320 act_values))
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]321{
322 // Validate output
323 validate(CLAccessor(_target), _reference, rel_tolerance_f32, 0.f, abs_tolerance_f32);
324}
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]325TEST_SUITE_END() // FP32
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]326
327TEST_SUITE(FP16)
328FIXTURE_DATA_TEST_CASE(RunSmall, CLGEMMMatrixMultiplyReshapedFixture<half>, framework::DatasetMode::ALL,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]329 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]330 m_values,
331 n_values),
332 k_values),
333 b_values),
334 m0_values_precommit),
335 n0_values_precommit),
336 k0_values_precommit),
337 v0_values_precommit),
338 h0_values_precommit),
339 i_values_lhs),
340 i_values_rhs),
341 framework::dataset::make("DataType", DataType::F16)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]342 a_values),
343 beta_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]344 broadcast_bias_values),
345 act_values))
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]346{
347 // Validate output
348 validate(CLAccessor(_target), _reference, rel_tolerance_f16, tolerance_num_f16);
349}
350
351FIXTURE_DATA_TEST_CASE(RunLarge, CLGEMMMatrixMultiplyReshapedFixture<half>, framework::DatasetMode::NIGHTLY,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]352 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]353 m_values,
354 n_values),
355 k_values),
356 b_values),
357 m0_values_nightly),
358 n0_values_nightly),
359 k0_values_nightly),
360 v0_values_nightly),
361 h0_values_nightly),
362 i_values_lhs),
363 i_values_rhs),
364 framework::dataset::make("DataType", DataType::F16)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]365 a_values),
366 beta_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]367 broadcast_bias_values),
368 act_values))
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]369{
370 // Validate output
371 validate(CLAccessor(_target), _reference, rel_tolerance_f16, tolerance_num_f16);
372}
373
374FIXTURE_DATA_TEST_CASE(RunSmall3D, CLGEMMMatrixMultiplyReshaped3DFixture<half>, framework::DatasetMode::ALL,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]375 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]376 m_w_values,
377 m_h_values),
378 n_values),
379 k_values),
380 b_values),
381 m0_values_precommit),
382 n0_values_precommit),
383 k0_values_precommit),
384 v0_values_precommit),
385 h0_values_precommit),
386 i_values_lhs),
387 i_values_rhs),
388 framework::dataset::make("DataType", DataType::F16)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]389 a_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]390 beta_values),
391 act_values))
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]392{
393 // Validate output
394 validate(CLAccessor(_target), _reference, rel_tolerance_f16, tolerance_num_f16);
395}
396
397FIXTURE_DATA_TEST_CASE(RunLarge3D, CLGEMMMatrixMultiplyReshaped3DFixture<half>, framework::DatasetMode::NIGHTLY,
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]398 combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(combine(
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]399 m_w_values,
400 m_h_values),
401 n_values),
402 k_values),
403 b_values),
404 m0_values_nightly),
405 n0_values_nightly),
406 k0_values_nightly),
407 v0_values_nightly),
408 h0_values_nightly),
409 i_values_lhs),
410 i_values_rhs),
411 framework::dataset::make("DataType", DataType::F16)),
Gian Marco Iodicee16c8902019-06-14 16:11:10 +0100[diff] [blame]412 a_values),
Gian Marco Iodiceca1f4602019-07-16 15:46:48 +0100[diff] [blame]413 beta_values),
414 act_values))
Gian Marco Iodice9382ab32018-12-17 15:12:07 +0000[diff] [blame]415{
416 // Validate output
417 validate(CLAccessor(_target), _reference, rel_tolerance_f16, tolerance_num_f16);
418}
419TEST_SUITE_END() // FP16
420TEST_SUITE_END() // Float
Gian Marco Iodiced1f54762019-07-19 09:54:47 +0100[diff] [blame]421TEST_SUITE_END() // GEMMMatrixMultiplyReshaped
Gian Marco Iodicebf9731e2018-12-12 10:18:04 +0000[diff] [blame]422TEST_SUITE_END() // CL
423} // namespace validation
424} // namespace test
425} // namespace arm_compute