xref: /aosp_15_r20/external/ComputeLibrary/tests/validation/CL/FillBorder.cpp (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1 /*
2  * Copyright (c) 2017-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/runtime/CL/CLScheduler.h"
25 #include "src/core/CL/kernels/CLFillBorderKernel.h"
26 #include "tests/CL/CLAccessor.h"
27 #include "tests/Globals.h"
28 #include "tests/datasets/BorderModeDataset.h"
29 #include "tests/datasets/ShapeDatasets.h"
30 #include "tests/framework/Macros.h"
31 #include "tests/framework/datasets/Datasets.h"
32 #include "tests/validation/Validation.h"
33 
34 namespace arm_compute
35 {
36 namespace test
37 {
38 namespace validation
39 {
40 TEST_SUITE(CL)
41 TEST_SUITE(FillBorder)
42 
43 // *INDENT-OFF*
44 // clang-format off
45 const auto PaddingSizesDataset = concat(concat(
46                                  framework::dataset::make("PaddingSize", PaddingSize{ 0 }),
47                                  framework::dataset::make("PaddingSize", PaddingSize{ 1, 0, 1, 2 })),
48                                  framework::dataset::make("PaddingSize", PaddingSize{ 10 }));
49 
50 const auto BorderSizesDataset  = framework::dataset::make("BorderSize", 0, 6);
51 
52 DATA_TEST_CASE(FillBorder, framework::DatasetMode::ALL, combine(combine(combine(combine(
53                datasets::SmallShapes(),
54                datasets::BorderModes()),
55                BorderSizesDataset),
56                PaddingSizesDataset),
57                framework::dataset::make("DataType", DataType::U8)),
58                shape, border_mode, size, padding, data_type)
59 // clang-format on
60 // *INDENT-ON*
61 {
62     BorderSize border_size{ static_cast<unsigned int>(size) };
63 
64     std::mt19937                           generator(library->seed());
65     std::uniform_int_distribution<uint8_t> distribution_u8(0, 255);
66     const uint8_t                          border_value = distribution_u8(generator);
67     const uint8_t                          tensor_value = distribution_u8(generator);
68 
69     // Create tensors
70     CLTensor src = create_tensor<CLTensor>(shape, data_type);
71 
72     src.info()->extend_padding(padding);
73 
74     // Allocate tensor
75     src.allocator()->allocate();
76 
77     // Check padding is as required
78     validate(src.info()->padding(), padding);
79 
80     // Fill tensor with constant value
81     std::uniform_int_distribution<uint8_t> distribution{ tensor_value, tensor_value };
82     library->fill(CLAccessor(src), distribution, 0);
83 
84     // Create and configure kernel
85     CLFillBorderKernel fill_border;
86     fill_border.configure(&src, border_size, border_mode, border_value);
87 
88     // Run kernel
89     fill_border.run(fill_border.window(), CLScheduler::get().queue());
90 
91     // Validate border
92     border_size.limit(padding);
93     validate(CLAccessor(src), border_size, border_mode, &border_value);
94 
95     // Validate tensor
96     validate(CLAccessor(src), &tensor_value);
97 }
98 
99 TEST_SUITE_END()
100 TEST_SUITE_END()
101 } // namespace validation
102 } // namespace test
103 } // namespace arm_compute
104