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/runtime/NEON/functions/NEPermute.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/PermuteFixture.h"
36
37 namespace arm_compute
38 {
39 namespace test
40 {
41 namespace validation
42 {
43 namespace
44 {
45 const auto PermuteVectors2 = framework::dataset::make("PermutationVector",
46 {
47 PermutationVector(0U, 1U),
48 PermutationVector(1U, 0U),
49 });
50 const auto PermuteVectors3 = framework::dataset::make("PermutationVector",
51 {
52 PermutationVector(2U, 0U, 1U),
53 PermutationVector(1U, 2U, 0U),
54 PermutationVector(0U, 1U, 2U),
55 PermutationVector(0U, 2U, 1U),
56 PermutationVector(1U, 0U, 2U),
57 PermutationVector(2U, 1U, 0U),
58 });
59 const auto PermuteVectors4 = framework::dataset::make("PermutationVector",
60 {
61 PermutationVector(3U, 2U, 0U, 1U),
62 PermutationVector(3U, 2U, 1U, 0U),
63 PermutationVector(2U, 3U, 1U, 0U),
64 PermutationVector(1U, 3U, 2U, 0U),
65 PermutationVector(3U, 1U, 2U, 0U),
66 PermutationVector(3U, 0U, 2U, 1U),
67 PermutationVector(0U, 3U, 2U, 1U)
68 });
69 const auto PermuteVectors = concat(concat(PermuteVectors2, PermuteVectors3), PermuteVectors4);
70 const auto PermuteParametersSmall = concat(concat(datasets::Small2DShapes(), datasets::Small3DShapes()), datasets::Small4DShapes()) * PermuteVectors;
71 const auto PermuteParametersLarge = datasets::Large4DShapes() * PermuteVectors;
72 } // namespace
73 TEST_SUITE(NEON)
TEST_SUITE(Permute)74 TEST_SUITE(Permute)
75
76 // *INDENT-OFF*
77 // clang-format off
78 DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip(
79 framework::dataset::make("InputInfo",{
80 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::U16), // permutation not supported
81 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::U16), // permutation not supported
82 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::U16), // permutation not supported
83 TensorInfo(TensorShape(1U, 7U), 1, DataType::U8), // invalid input size
84 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::U16), // valid
85 TensorInfo(TensorShape(27U, 13U, 37U, 2U), 1, DataType::F32), // valid
86 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::U16), // permutation not supported
87 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::S16), // permutation not supported
88 TensorInfo(TensorShape(27U, 13U, 37U, 2U), 1, DataType::F32), // permutation not supported
89 TensorInfo(TensorShape(27U, 13U, 37U, 2U), 1, DataType::F32), // permutation not supported
90 TensorInfo(TensorShape(27U, 13U, 37U, 2U), 1, DataType::F32) // permutation not supported
91
92 }),
93 framework::dataset::make("OutputInfo", {
94 TensorInfo(TensorShape(5U, 7U, 7U, 3U), 1, DataType::U16),
95 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::U16),
96 TensorInfo(TensorShape(7U, 7U, 5U, 3U), 1, DataType::U16),
97 TensorInfo(TensorShape(5U, 7U), 1, DataType::U8),
98 TensorInfo(TensorShape(5U, 7U, 7U, 3U), 1, DataType::U16),
99 TensorInfo(TensorShape(13U, 37U, 27U, 2U), 1, DataType::F32),
100 TensorInfo(TensorShape(5U, 7U, 7U, 3U), 1, DataType::U16),
101 TensorInfo(TensorShape(3U, 5U, 7U, 7U), 1, DataType::S16),
102 TensorInfo(TensorShape(13U, 37U, 27U, 2U), 1, DataType::F32),
103 TensorInfo(TensorShape(37U, 2U, 13U, 27U), 1, DataType::F32),
104 TensorInfo(TensorShape(37U, 2U, 13U, 27U), 1, DataType::F32)
105
106 })),
107 framework::dataset::make("PermutationVector", {
108 PermutationVector(2U, 1U, 0U),
109 PermutationVector(2U, 2U, 1U),
110 PermutationVector(1U, 1U, 1U),
111 PermutationVector(2U, 0U, 1U),
112 PermutationVector(2U, 0U, 1U),
113 PermutationVector(1U, 2U, 0U),
114 PermutationVector(3U, 2U, 0U, 1U),
115 PermutationVector(3U, 2U, 0U, 1U),
116 PermutationVector(2U, 3U, 1U, 0U),
117 PermutationVector(2U, 3U, 1U, 0U),
118 PermutationVector(0U, 0U, 0U, 1000U)
119 })),
120 framework::dataset::make("Expected", { true, false, false, false, true, true, false,true, false, true, false })),
121 input_info, output_info, perm_vect, expected)
122 {
123 ARM_COMPUTE_EXPECT(bool(NEPermute::validate(&input_info.clone()->set_is_resizable(false), &output_info.clone()->set_is_resizable(false), perm_vect)) == expected, framework::LogLevel::ERRORS);
124 }
125 // clang-format on
126 // *INDENT-ON*
127
128 template <typename T>
129 using NEPermuteFixture = PermuteValidationFixture<Tensor, Accessor, NEPermute, T>;
130
131 TEST_SUITE(U8)
132 FIXTURE_DATA_TEST_CASE(RunSmall, NEPermuteFixture<uint8_t>, framework::DatasetMode::PRECOMMIT,
133 PermuteParametersSmall * framework::dataset::make("DataType", DataType::U8))
134 {
135 // Validate output
136 validate(Accessor(_target), _reference);
137 }
138
139 FIXTURE_DATA_TEST_CASE(RunLarge, NEPermuteFixture<uint8_t>, framework::DatasetMode::NIGHTLY,
140 PermuteParametersLarge * framework::dataset::make("DataType", DataType::U8))
141 {
142 // Validate output
143 validate(Accessor(_target), _reference);
144 }
145 TEST_SUITE_END()
146
TEST_SUITE(U16)147 TEST_SUITE(U16)
148 FIXTURE_DATA_TEST_CASE(RunSmall, NEPermuteFixture<uint16_t>, framework::DatasetMode::PRECOMMIT,
149 PermuteParametersSmall * framework::dataset::make("DataType", DataType::U16))
150 {
151 // Validate output
152 validate(Accessor(_target), _reference);
153 }
154 FIXTURE_DATA_TEST_CASE(RunLarge, NEPermuteFixture<uint16_t>, framework::DatasetMode::NIGHTLY,
155 PermuteParametersLarge * framework::dataset::make("DataType", DataType::U16))
156 {
157 // Validate output
158 validate(Accessor(_target), _reference);
159 }
160 TEST_SUITE_END()
161
TEST_SUITE(U32)162 TEST_SUITE(U32)
163 FIXTURE_DATA_TEST_CASE(RunSmall, NEPermuteFixture<uint32_t>, framework::DatasetMode::PRECOMMIT,
164 PermuteParametersSmall * framework::dataset::make("DataType", DataType::U32))
165 {
166 // Validate output
167 validate(Accessor(_target), _reference);
168 }
169 FIXTURE_DATA_TEST_CASE(RunLarge, NEPermuteFixture<uint32_t>, framework::DatasetMode::NIGHTLY,
170 PermuteParametersLarge * framework::dataset::make("DataType", DataType::U32))
171 {
172 // Validate output
173 validate(Accessor(_target), _reference);
174 }
175 TEST_SUITE_END()
176
177 TEST_SUITE_END()
178 TEST_SUITE_END()
179 } // namespace validation
180 } // namespace test
181 } // namespace arm_compute
182