blob: a85a0e7e98387425ab491bcbe49d28bc871ab114 [file] [log] [blame]
Frank Lei57a150a2017-12-19 10:14:57 +08001/*
Matthew Bentham758b5ba2020-03-05 23:37:48 +00002 * Copyright (c) 2016-2020 ARM Limited.
Frank Lei57a150a2017-12-19 10:14:57 +08003 *
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/GLES_COMPUTE/kernels/GCScaleKernel.h"
25
26#include "arm_compute/core/AccessWindowStatic.h"
27#include "arm_compute/core/Error.h"
28#include "arm_compute/core/GLES_COMPUTE/GCHelpers.h"
29#include "arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h"
30#include "arm_compute/core/GLES_COMPUTE/IGCKernel.h"
31#include "arm_compute/core/GLES_COMPUTE/IGCTensor.h"
32#include "arm_compute/core/GLES_COMPUTE/OpenGLES.h"
33#include "arm_compute/core/Helpers.h"
34#include "arm_compute/core/TensorInfo.h"
35#include "arm_compute/core/Validate.h"
Matthew Bentham758b5ba2020-03-05 23:37:48 +000036#include "support/StringSupport.h"
Frank Lei57a150a2017-12-19 10:14:57 +080037
38#include <set>
39#include <string>
40
41using namespace arm_compute;
42
43BorderSize GCScaleKernel::border_size() const
44{
45 return BorderSize(1);
46}
47
48void GCScaleKernel::configure(const IGCTensor *input, IGCTensor *output, InterpolationPolicy policy, bool border_undefined, SamplingPolicy sampling_policy)
49{
50 ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input, 1, DataType::F16);
51 ARM_COMPUTE_ERROR_ON_NULLPTR(output);
52 ARM_COMPUTE_ERROR_ON_MISMATCHING_DATA_TYPES(input, output);
53 ARM_COMPUTE_ERROR_ON(output == input);
54 ARM_COMPUTE_ERROR_ON(policy != InterpolationPolicy::NEAREST_NEIGHBOR);
Frank Lei57a150a2017-12-19 10:14:57 +080055
56 _input = input;
57 _output = output;
58
59 // Compute the ratio between source width/height and destination width/height
60 const auto wr = static_cast<float>(input->info()->dimension(0)) / static_cast<float>(output->info()->dimension(0));
61 const auto hr = static_cast<float>(input->info()->dimension(1)) / static_cast<float>(output->info()->dimension(1));
62
63 // Compute actual border size
64 BorderSize border = border_undefined ? BorderSize(0) : border_size();
65
66 // Area interpolation behaves as Nearest Neighbour in case of up-sampling
67 if(policy == InterpolationPolicy::AREA && wr <= 1.f && hr <= 1.f)
68 {
69 policy = InterpolationPolicy::NEAREST_NEIGHBOR;
70 }
71 else
72 {
73 ARM_COMPUTE_ERROR_ON(policy == InterpolationPolicy::AREA);
74 }
75
76 // Create kernel
77 std::set<std::string> build_opts;
78 build_opts.emplace("#define LOCAL_SIZE_X " + support::cpp11::to_string(1));
79 build_opts.emplace("#define LOCAL_SIZE_Y " + support::cpp11::to_string(1));
80 build_opts.emplace("#define LOCAL_SIZE_Z " + support::cpp11::to_string(1));
81
82 build_opts.emplace("#define DATA_TYPE_FP16");
83 build_opts.emplace("#define BORDER_SIZE " + support::cpp11::to_string(border.right));
Michalis Spyroud4733862019-07-09 14:21:06 +010084 if(sampling_policy == SamplingPolicy::TOP_LEFT)
85 {
86 build_opts.emplace("#define SAMPLING_POLICY_TOP_LEFT");
87 }
88 else
89 {
90 build_opts.emplace("#define SAMPLING_POLICY_CENTER");
91 }
Frank Lei57a150a2017-12-19 10:14:57 +080092
93 // Configure kernel window
94 unsigned int num_elems_processed_per_iteration = 4;
95 unsigned int input_width_alignment = 2;
96
97 // performance optimization for 2x upscaling with no border
98 if((fabs(wr - 0.5) < 1e-6) && (fabs(hr - 0.5) < 1e-6) && border_undefined)
99 {
100 num_elems_processed_per_iteration = 8;
101 input_width_alignment = 4;
102 build_opts.emplace("#define SCALE_NEAREST_8X");
103 }
104 else
105 {
106 build_opts.emplace("#define SCALE_NEAREST_GENERIC");
107 }
108
109 std::string interpolation_name = string_from_interpolation_policy(policy); // NOLINT
110 std::transform(interpolation_name.begin(), interpolation_name.end(), interpolation_name.begin(), ::tolower);
111 std::string kernel_name = "scale_" + interpolation_name;
112 _kernel = GCKernelLibrary::get().create_kernel(kernel_name, build_opts);
113
114 Window win = calculate_max_window(*output->info(), Steps(num_elems_processed_per_iteration));
115
116 const ValidRegion &input_valid_region = input->info()->valid_region();
117
118 const int total_width = border.left + input_valid_region.anchor[0] + input_valid_region.shape[0] + border.right;
119 const int padding_right = ceil_to_multiple(total_width, input_width_alignment) - border.left - input_valid_region.anchor[0] - input_valid_region.shape[0];
120
121 // Reads can occur within the valid region of the input
122 AccessWindowStatic input_access(input->info(),
123 input_valid_region.anchor[0] - border.left, input_valid_region.anchor[1] - border.top,
124 input_valid_region.anchor[0] + input_valid_region.shape[0] + padding_right,
125 input_valid_region.anchor[1] + input_valid_region.shape[1] + border.bottom);
126
127 AccessWindowHorizontal output_access(output->info(), 0, num_elems_processed_per_iteration);
128
129 update_window_and_padding(win, input_access, output_access);
130
131 output_access.set_valid_region(win, calculate_valid_region_scale(*(input->info()),
132 output->info()->tensor_shape(),
133 policy,
Diego Lopez Recas00854292018-02-22 13:08:01 +0000134 sampling_policy,
Frank Lei57a150a2017-12-19 10:14:57 +0800135 border_undefined));
136
137 IGCKernel::configure(win);
138
Frank Lei4406fd62018-02-01 14:47:14 +0800139 unsigned int idx = 2 * num_arguments_per_3D_tensor(); //Skip the tensor parameters
Frank Lei57a150a2017-12-19 10:14:57 +0800140 _kernel.set_argument<float>(idx++, static_cast<float>(input->info()->dimension(0)));
141 _kernel.set_argument<float>(idx++, static_cast<float>(input->info()->dimension(1)));
Frank Lei1bfc7842018-01-04 13:02:40 +0800142 _kernel.set_argument<float>(idx++, wr);
143 _kernel.set_argument<float>(idx++, hr);
Frank Lei57a150a2017-12-19 10:14:57 +0800144}
Frank Lei4406fd62018-02-01 14:47:14 +0800145
146void GCScaleKernel::run(const Window &window)
147{
148 ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this);
149 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(IKernel::window(), window);
150
151 _kernel.use();
152
153 _output->set_needs_shifting(true);
154
155 Window slice = window.first_slice_window_3D();
156 Window slice_in = window.first_slice_window_3D();
157
158 slice.shift(Window::DimX, -(_output->info()->padding()).left);
159
160 do
161 {
162 unsigned int idx = 0;
163 add_3D_tensor_argument(idx, _input, 1, slice_in);
164 add_3D_tensor_argument(idx, _output, 2, slice);
165 _kernel.update_shader_params();
166 enqueue(*this, slice);
167 }
168 while(window.slide_window_slice_3D(slice) && window.slide_window_slice_3D(slice_in));
169}