blob: 0b873945d30ef2b7aed65ce12ede12c4a69b01db [file] [log] [blame]
Michalis Spyrou7930db42018-11-22 17:36:28 +00001/*
2 * Copyright (c) 2018 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/runtime/CL/CLTensor.h"
26#include "arm_compute/runtime/CL/CLTensorAllocator.h"
27#include "arm_compute/runtime/CL/functions/CLArgMinMaxLayer.h"
28
29#include "tests/CL/CLAccessor.h"
30#include "tests/datasets/ShapeDatasets.h"
31#include "tests/datasets/SplitDataset.h"
32#include "tests/framework/Asserts.h"
33#include "tests/framework/Macros.h"
34#include "tests/validation/Validation.h"
35#include "tests/validation/fixtures/ArgMinMaxFixture.h"
36
37namespace arm_compute
38{
39namespace test
40{
41namespace validation
42{
43TEST_SUITE(CL)
44TEST_SUITE(ArgMinMax)
45
46// *INDENT-OFF*
47// clang-format off
48DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip(zip(
49 framework::dataset::make("InputInfo", { TensorInfo(TensorShape(27U, 3U, 16U, 2U), 1, DataType::F32), // Invalid axis
50 TensorInfo(TensorShape(27U, 3U, 16U, 2U), 1, DataType::F32), // Invalid output shape
51 TensorInfo(TensorShape(32U, 16U, 16U, 2U), 1, DataType::F32),
52 TensorInfo(TensorShape(32U, 16U, 16U, 2U), 1, DataType::F32) // Invalid operation
53 }),
54 framework::dataset::make("OutputInfo", { TensorInfo(TensorShape(27U, 3U, 1U, 2U), 1, DataType::F32),
55 TensorInfo(TensorShape(27U, 3U, 1U, 2U), 1, DataType::F32),
56 TensorInfo(TensorShape(32U, 16U, 1U, 2U), 1, DataType::U32),
57 TensorInfo(TensorShape(32U, 16U, 1U, 2U), 1, DataType::F32)
58 })),
59 framework::dataset::make("Axis", { 4, 0, 2, 0 })),
60 framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MAX, ReductionOperation::ARG_IDX_MAX, ReductionOperation::ARG_IDX_MAX, ReductionOperation::MEAN_SUM })),
61 framework::dataset::make("Expected", { false, false, true, false })),
62 input_info, output_info, axis, operation, expected)
63{
64 const Status status = CLArgMinMaxLayer::validate(&input_info.clone()->set_is_resizable(false), axis, &output_info.clone()->set_is_resizable(false), operation);
65 ARM_COMPUTE_EXPECT(bool(status) == expected, framework::LogLevel::ERRORS);
66}
67// clang-format on
68// *INDENT-ON*
69
70DATA_TEST_CASE(Configuration,
71 framework::DatasetMode::ALL,
72 combine(datasets::SmallShapes(), framework::dataset::make("DataType", { DataType::F16, DataType::F32 })),
73 shape, data_type)
74{
75 // Create tensors
76 CLTensor ref_src = create_tensor<CLTensor>(shape, data_type);
77 CLTensor dst;
78
79 // Create and Configure function
80 CLArgMinMaxLayer arg_min_max_layer;
81 arg_min_max_layer.configure(&ref_src, 1, &dst, ReductionOperation::ARG_IDX_MAX);
82
83 // Validate valid region
84 TensorShape output_shape = shape;
85 output_shape.set(1, 1);
86 const ValidRegion valid_region = shape_to_valid_region(output_shape);
87 validate(dst.info()->valid_region(), valid_region);
88}
89
90template <typename T>
91using CLArgMinMaxValidationFixture = ArgMinMaxValidationFixture<CLTensor, CLAccessor, CLArgMinMaxLayer, T>;
92
93TEST_SUITE(Float)
94TEST_SUITE(FP16)
95FIXTURE_DATA_TEST_CASE(RunSmall,
96 CLArgMinMaxValidationFixture<half>,
97 framework::DatasetMode::PRECOMMIT,
98 combine(combine(combine(datasets::Small4DShapes(), framework::dataset::make("DataType", DataType::F16)), framework::dataset::make("Axis", { 0, 1, 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_IDX_MAX })))
99{
100 // Validate output
101 validate(CLAccessor(_target), _reference);
102}
103
104FIXTURE_DATA_TEST_CASE(RunLarge,
105 CLArgMinMaxValidationFixture<half>,
106 framework::DatasetMode::NIGHTLY,
107 combine(combine(combine(datasets::Large4DShapes(), framework::dataset::make("DataType", DataType::F16)), framework::dataset::make("Axis", { 0, 1, 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_IDX_MAX })))
108{
109 // Validate output
110 validate(CLAccessor(_target), _reference);
111}
112TEST_SUITE_END() // FP16
113
114TEST_SUITE(FP32)
115FIXTURE_DATA_TEST_CASE(RunSmall,
116 CLArgMinMaxValidationFixture<float>,
117 framework::DatasetMode::PRECOMMIT,
118 combine(combine(combine(datasets::Small4DShapes(), framework::dataset::make("DataType", DataType::F32)), framework::dataset::make("Axis", { 0, 1, 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_IDX_MAX })))
119{
120 // Validate output
121 validate(CLAccessor(_target), _reference);
122}
123
124FIXTURE_DATA_TEST_CASE(RunLarge,
125 CLArgMinMaxValidationFixture<float>,
126 framework::DatasetMode::NIGHTLY,
127 combine(combine(combine(datasets::Large4DShapes(), framework::dataset::make("DataType", DataType::F32)), framework::dataset::make("Axis", { 0, 1, 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_IDX_MAX })))
128{
129 // Validate output
130 validate(CLAccessor(_target), _reference);
131}
132TEST_SUITE_END() // FP32
133TEST_SUITE_END() // Float
134TEST_SUITE_END() // ArgMinMax
135TEST_SUITE_END() // CL
136} // namespace validation
137} // namespace test
138} // namespace arm_compute