xref: /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ElementwisePower.cpp (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1 /*
2  * Copyright (c) 2019-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 #include "arm_compute/core/Types.h"
25 #include "arm_compute/runtime/NEON/functions/NEElementwiseOperations.h"
26 #include "arm_compute/runtime/Tensor.h"
27 #include "arm_compute/runtime/TensorAllocator.h"
28 #include "tests/NEON/Accessor.h"
29 #include "tests/PaddingCalculator.h"
30 #include "tests/datasets/ShapeDatasets.h"
31 #include "tests/framework/Asserts.h"
32 #include "tests/framework/Macros.h"
33 #include "tests/framework/datasets/Datasets.h"
34 #include "tests/validation/Validation.h"
35 #include "tests/validation/fixtures/ElementwiseOperationsFixture.h"
36 
37 namespace arm_compute
38 {
39 namespace test
40 {
41 namespace validation
42 {
43 namespace
44 {
45 RelativeTolerance<float> tolerance_fp32(0.001f);
46 /** Input data sets **/
47 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
48 RelativeTolerance<half> tolerance_fp16(static_cast<half>(0.01f));
49 const auto              ElementwisePowerFP16Dataset = combine(combine(framework::dataset::make("DataType", DataType::F16), framework::dataset::make("DataType", DataType::F16)),
50                                                               framework::dataset::make("DataType", DataType::F16));
51 #endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
52 const auto ElementwisePowerFP32Dataset = combine(combine(framework::dataset::make("DataType", DataType::F32), framework::dataset::make("DataType", DataType::F32)),
53                                                  framework::dataset::make("DataType", DataType::F32));
54 const auto InPlaceDataSet    = framework::dataset::make("InPlace", { false, true });
55 const auto OutOfPlaceDataSet = framework::dataset::make("InPlace", { false });
56 } // namespace
57 
58 TEST_SUITE(NEON)
59 TEST_SUITE(ElementwisePower)
60 
61 template <typename T>
62 using NEElementwisePowerFixture = ElementwisePowerValidationFixture<Tensor, Accessor, NEElementwisePower, T>;
63 
64 // *INDENT-OFF*
65 // clang-format off
66 DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip(
67                framework::dataset::make("Input1Info", { TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),
68                                                         TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),
69                                                         TensorInfo(TensorShape(27U, 13U, 2U), 1, DataType::F32),
70                                                         TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),     // Invalid data type combination
71                                                         TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),     // Mismatching shapes
72                                                       }),
73                framework::dataset::make("Input2Info",{ TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),
74                                                        TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),
75                                                        TensorInfo(TensorShape(27U, 13U, 2U), 1, DataType::F32),
76                                                        TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::S32),
77                                                        TensorInfo(TensorShape(48U, 11U, 2U), 1, DataType::F32),
78                                                      })),
79                framework::dataset::make("OutputInfo",{ TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),
80                                                        TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),
81                                                        TensorInfo(TensorShape(27U, 13U, 2U), 1, DataType::F32),
82                                                        TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32),
83                                                        TensorInfo(TensorShape(48U, 11U, 2U), 1, DataType::F32),
84                                                      })),
85                framework::dataset::make("Expected", { true, true, true, false, false})),
86                input1_info, input2_info, output_info, expected)
87 {
88     ARM_COMPUTE_EXPECT(bool(NEElementwisePower::validate(&input1_info.clone()->set_is_resizable(false), &input2_info.clone()->set_is_resizable(false), &output_info.clone()->set_is_resizable(false))) == expected, framework::LogLevel::ERRORS);
89 }
90 // clang-format on
91 // *INDENT-ON*
92 
93 TEST_SUITE(Float)
94 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
TEST_SUITE(F16)95 TEST_SUITE(F16)
96 FIXTURE_DATA_TEST_CASE(RunSmall, NEElementwisePowerFixture<half>, framework::DatasetMode::ALL, combine(combine(datasets::SmallShapes(), ElementwisePowerFP16Dataset),
97                                                                                                        InPlaceDataSet))
98 {
99     // Validate output
100     validate(Accessor(_target), _reference, tolerance_fp16, 0.01);
101 }
102 TEST_SUITE_END() // F16
103 #endif           /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
104 
TEST_SUITE(F32)105 TEST_SUITE(F32)
106 
107 FIXTURE_DATA_TEST_CASE(RunSmall, NEElementwisePowerFixture<float>, framework::DatasetMode::ALL, combine(combine(datasets::SmallShapes(), ElementwisePowerFP32Dataset),
108                                                                                                         InPlaceDataSet))
109 {
110     // Validate output
111     validate(Accessor(_target), _reference, tolerance_fp32, 0.01);
112 }
113 
FIXTURE_DATA_TEST_CASE(RunLarge,NEElementwisePowerFixture<float>,framework::DatasetMode::NIGHTLY,combine (combine (datasets::LargeShapes (),ElementwisePowerFP32Dataset),InPlaceDataSet))114 FIXTURE_DATA_TEST_CASE(RunLarge, NEElementwisePowerFixture<float>, framework::DatasetMode::NIGHTLY, combine(combine(datasets::LargeShapes(), ElementwisePowerFP32Dataset),
115                                                                                                             InPlaceDataSet))
116 {
117     // Validate output
118     validate(Accessor(_target), _reference, tolerance_fp32, 0.01);
119 }
120 
121 template <typename T>
122 using NEElementwisePowerBroadcastFixture = ElementwisePowerBroadcastValidationFixture<Tensor, Accessor, NEElementwisePower, T>;
123 
FIXTURE_DATA_TEST_CASE(RunSmallBroadcast,NEElementwisePowerBroadcastFixture<float>,framework::DatasetMode::ALL,combine (combine (datasets::SmallShapesBroadcast (),ElementwisePowerFP32Dataset),OutOfPlaceDataSet))124 FIXTURE_DATA_TEST_CASE(RunSmallBroadcast, NEElementwisePowerBroadcastFixture<float>, framework::DatasetMode::ALL, combine(combine(datasets::SmallShapesBroadcast(),
125                        ElementwisePowerFP32Dataset),
126                        OutOfPlaceDataSet))
127 {
128     // Validate output
129     validate(Accessor(_target), _reference, tolerance_fp32, 0.01);
130 }
FIXTURE_DATA_TEST_CASE(RunTinyBroadcastInPlace,NEElementwisePowerBroadcastFixture<float>,framework::DatasetMode::ALL,combine (combine (datasets::TinyShapesBroadcastInplace (),ElementwisePowerFP32Dataset),InPlaceDataSet))131 FIXTURE_DATA_TEST_CASE(RunTinyBroadcastInPlace, NEElementwisePowerBroadcastFixture<float>, framework::DatasetMode::ALL, combine(combine(datasets::TinyShapesBroadcastInplace(),
132                        ElementwisePowerFP32Dataset),
133                        InPlaceDataSet))
134 {
135     // Validate output
136     validate(Accessor(_target), _reference, tolerance_fp32, 0.01);
137 }
FIXTURE_DATA_TEST_CASE(RunLargeBroadcast,NEElementwisePowerBroadcastFixture<float>,framework::DatasetMode::NIGHTLY,combine (combine (datasets::LargeShapesBroadcast (),ElementwisePowerFP32Dataset),OutOfPlaceDataSet))138 FIXTURE_DATA_TEST_CASE(RunLargeBroadcast, NEElementwisePowerBroadcastFixture<float>, framework::DatasetMode::NIGHTLY, combine(combine(datasets::LargeShapesBroadcast(),
139                        ElementwisePowerFP32Dataset),
140                        OutOfPlaceDataSet))
141 {
142     // Validate output
143     validate(Accessor(_target), _reference, tolerance_fp32, 0.01);
144 }
145 TEST_SUITE_END() // F32
146 TEST_SUITE_END() // Float
147 
148 TEST_SUITE_END() // ElementwisePower
149 TEST_SUITE_END() // Neon
150 } // namespace validation
151 } // namespace test
152 } // namespace arm_compute
153