xref: /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/Reverse.cpp (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1 /*
2  * Copyright (c) 2018-2021 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 
25 #include "arm_compute/core/Types.h"
26 #include "arm_compute/runtime/NEON/functions/NEReverse.h"
27 #include "arm_compute/runtime/Tensor.h"
28 #include "arm_compute/runtime/TensorAllocator.h"
29 #include "tests/NEON/Accessor.h"
30 #include "tests/PaddingCalculator.h"
31 #include "tests/datasets/ShapeDatasets.h"
32 #include "tests/framework/Asserts.h"
33 #include "tests/framework/Macros.h"
34 #include "tests/framework/datasets/Datasets.h"
35 #include "tests/validation/Validation.h"
36 #include "tests/validation/fixtures/ReverseFixture.h"
37 
38 namespace arm_compute
39 {
40 namespace test
41 {
42 namespace validation
43 {
44 namespace
45 {
46 auto run_small_dataset = combine(datasets::SmallShapes(), datasets::Tiny1DShapes());
47 auto run_large_dataset = combine(datasets::LargeShapes(), datasets::Tiny1DShapes());
48 
49 } // namespace
50 TEST_SUITE(NEON)
TEST_SUITE(Reverse)51 TEST_SUITE(Reverse)
52 
53 // *INDENT-OFF*
54 // clang-format off
55 DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip(
56         framework::dataset::make("InputInfo", { TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::S8), // Invalid axis datatype
57                                             TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::U8), // Invalid axis shape
58                                             TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::U8), // Invalid axis length (> 4)
59                                             TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::U8), // Mismatching shapes
60                                             TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::U8),
61                                             TensorInfo(TensorShape(2U), 1, DataType::U8),
62         }),
63         framework::dataset::make("OutputInfo", { TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::S8),
64                                             TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::U8),
65                                             TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::U8),
66                                             TensorInfo(TensorShape(2U, 13U, 2U), 1, DataType::U8),
67                                             TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::U8),
68                                             TensorInfo(TensorShape(2U), 1, DataType::U8),
69         })),
70         framework::dataset::make("AxisInfo", { TensorInfo(TensorShape(3U), 1, DataType::U8),
71                                            TensorInfo(TensorShape(2U, 10U), 1, DataType::U32),
72                                            TensorInfo(TensorShape(8U), 1, DataType::U32),
73                                            TensorInfo(TensorShape(2U), 1, DataType::U32),
74                                            TensorInfo(TensorShape(2U), 1, DataType::U32),
75                                            TensorInfo(TensorShape(2U), 1, DataType::U32),
76         })),
77         framework::dataset::make("Expected", { false, false, false, false, true, true})),
78         src_info, dst_info, axis_info, expected)
79 {
80     Status s = NEReverse::validate(&src_info.clone()->set_is_resizable(false),
81                                   &dst_info.clone()->set_is_resizable(false),
82                                   &axis_info.clone()->set_is_resizable(false));
83     ARM_COMPUTE_EXPECT(bool(s) == expected, framework::LogLevel::ERRORS);
84 }
85 // clang-format on
86 // *INDENT-ON*
87 
88 template <typename T>
89 using NEReverseFixture = ReverseValidationFixture<Tensor, Accessor, NEReverse, T>;
90 
91 TEST_SUITE(Float)
92 
93 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
TEST_SUITE(F16)94 TEST_SUITE(F16)
95 FIXTURE_DATA_TEST_CASE(RunSmall,
96                        NEReverseFixture<half>,
97                        framework::DatasetMode::PRECOMMIT,
98                        combine(run_small_dataset, framework::dataset::make("DataType", DataType::F16)))
99 {
100     // Validate output
101     validate(Accessor(_target), _reference);
102 }
103 
104 FIXTURE_DATA_TEST_CASE(RunLarge,
105                        NEReverseFixture<half>,
106                        framework::DatasetMode::NIGHTLY,
107                        combine(run_large_dataset, framework::dataset::make("DataType", DataType::F16)))
108 {
109     // Validate output
110     validate(Accessor(_target), _reference);
111 }
112 TEST_SUITE_END() // F16
113 #endif           /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
114 
TEST_SUITE(FP32)115 TEST_SUITE(FP32)
116 FIXTURE_DATA_TEST_CASE(RunSmall,
117                        NEReverseFixture<float>,
118                        framework::DatasetMode::PRECOMMIT,
119                        combine(run_small_dataset, framework::dataset::make("DataType", DataType::F32)))
120 {
121     // Validate output
122     validate(Accessor(_target), _reference);
123 }
124 
125 FIXTURE_DATA_TEST_CASE(RunLarge,
126                        NEReverseFixture<float>,
127                        framework::DatasetMode::NIGHTLY,
128                        combine(run_large_dataset, framework::dataset::make("DataType", DataType::F32)))
129 {
130     // Validate output
131     validate(Accessor(_target), _reference);
132 }
133 TEST_SUITE_END() // F32
TEST_SUITE_END()134 TEST_SUITE_END() // Float
135 
136 TEST_SUITE(Quantized)
137 TEST_SUITE(QASYMM8)
138 FIXTURE_DATA_TEST_CASE(RunSmall,
139                        NEReverseFixture<uint8_t>,
140                        framework::DatasetMode::PRECOMMIT,
141                        combine(run_small_dataset, framework::dataset::make("DataType", DataType::QASYMM8)))
142 {
143     // Validate output
144     validate(Accessor(_target), _reference);
145 }
146 
147 FIXTURE_DATA_TEST_CASE(RunLarge,
148                        NEReverseFixture<uint8_t>,
149                        framework::DatasetMode::NIGHTLY,
150                        combine(run_large_dataset, framework::dataset::make("DataType", DataType::QASYMM8)))
151 {
152     // Validate output
153     validate(Accessor(_target), _reference);
154 }
155 TEST_SUITE_END() // QASYMM8
156 TEST_SUITE_END() // Quantized
157 
158 TEST_SUITE_END() // Reverse
159 TEST_SUITE_END() // Neon
160 } // namespace validation
161 } // namespace test
162 } // namespace arm_compute
163