blob: 846c712cebe1f0f4de04cec362c7347e43e3e0b9 [file] [log] [blame]
/*
* Copyright (c) 2022-2023 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to
* deal in the Software without restriction, including without limitation the
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
* sell copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include "ClTemplateResize.h"
#include "arm_compute/core/Utils.h"
#include "arm_compute/core/utils/helpers/AdjustVecSize.h"
#include "arm_compute/core/utils/StringUtils.h"
#include "src/core/helpers/WindowHelpers.h"
#include "src/core/utils/ScaleUtils.h"
#include "src/dynamic_fusion/sketch/gpu/GpuKernelComponentGroup.h"
namespace arm_compute
{
namespace experimental
{
namespace dynamic_fusion
{
ClTemplateResize::ClTemplateResize(ComponentId id,
const ArgumentPack<ITensorInfo> &tensors,
const ClTemplateResize::Attributes &attributes)
: IGpuTemplateComponentWriter{id, tensors}, _src{}, _dst{}, _attributes{attributes}
{
_src = this->tensors().get_const_tensor(TensorType::ACL_SRC_0);
_dst = this->tensors().get_const_tensor(TensorType::ACL_DST_0);
ARM_COMPUTE_ERROR_ON_NULLPTR(_src, _dst);
}
std::string ClTemplateResize::get_name() const
{
return _attributes.interpolation_policy() == InterpolationPolicy::BILINEAR ? "resize_bilinear" : "resize_nearest";
}
std::string ClTemplateResize::get_component_code(const IGpuTemplateComponentWriter::ComponentGroup &comp_group) const
{
ARM_COMPUTE_UNUSED(comp_group);
std::string code = R"_(
//------------------ START KERNEL {{meta_kernel_id}} ---------------------
TILE(uint, 1, 1, g_dst_indirect_y);
{
const int yo = g_ind_2 % {{arg_dst}}_h;
const int bout = g_ind_2 / {{arg_dst}}_h;
)_";
if (_attributes.interpolation_policy() == InterpolationPolicy::NEAREST_NEIGHBOR)
{
if (_attributes.sampling_policy() == SamplingPolicy::TOP_LEFT)
{
code += R"_(
float xi_f = (g_ind_1 * {{SCALE_X}});
float yi_f = (yo * {{SCALE_Y}});
)_";
}
else
{
code += R"_(
float xi_f = ((g_ind_1 + 0.5f) * {{SCALE_X}});
float yi_f = ((yo + 0.5f) * {{SCALE_Y}});
)_";
}
if (_attributes.align_corners())
{
code += R"_(
xi_f = round(xi_f);
yi_f = round(yi_f);
)_";
}
code += R"_(
const int xi0 = clamp((int)xi_f, 0, (int){{src}}_w - 1);
const int yi0 = clamp((int)yi_f, 0, (int){{src}}_h - 1);
T_LOAD_NHWC_WITH_DILATION({{SRC_DATA_TYPE}}, 1, 1, N0, {{SRC_TENSOR_TYPE}}, {{src}}, bout, yi0, xi0, g_ind_0, {{src}}_w, {{src}}_h, 1, 1, false, {{dst}});
)_";
}
else if (_attributes.interpolation_policy() == InterpolationPolicy::BILINEAR)
{
if (_attributes.sampling_policy() == SamplingPolicy::TOP_LEFT)
{
code += R"_(
float xi_f = (g_ind_1 * {{SCALE_X}});
float yi_f = (yo * {{SCALE_Y}});
)_";
}
else
{
code += R"_(
float xi_f = ((g_ind_1 + 0.5f) * {{SCALE_X}} - 0.5f);
float yi_f = ((yo + 0.5f) * {{SCALE_Y}} - 0.5f);
)_";
}
code += R"_(
const int xi = (int)floor(xi_f);
const int yi = (int)floor(yi_f);
TILE({{SRC_DATA_TYPE}}, 1, N0, in00);
TILE({{SRC_DATA_TYPE}}, 1, N0, in01);
TILE({{SRC_DATA_TYPE}}, 1, N0, in10);
TILE({{SRC_DATA_TYPE}}, 1, N0, in11);
in00[0].v = {{CONSTANT_VALUE}};
in01[0].v = {{CONSTANT_VALUE}};
in10[0].v = {{CONSTANT_VALUE}};
in11[0].v = {{CONSTANT_VALUE}};
const int xi0 = clamp(xi, 0, (int){{src}}_w - 1);
const int yi0 = clamp(yi, 0, (int){{src}}_h - 1);
const int xi1 = clamp(xi + 1, 0, (int){{src}}_w - 1);
const int yi1 = clamp(yi + 1, 0, (int){{src}}_h - 1);
T_LOAD_NHWC_WITH_DILATION({{SRC_DATA_TYPE}}, 1, 1, N0, {{SRC_TENSOR_TYPE}}, {{src}}, bout, yi0, xi0, g_ind_0, {{src}}_w, {{src}}_h, 1, 1, false, in00);
T_LOAD_NHWC_WITH_DILATION({{SRC_DATA_TYPE}}, 1, 1, N0, {{SRC_TENSOR_TYPE}}, {{src}}, bout, yi0, xi1, g_ind_0, {{src}}_w, {{src}}_h, 1, 1, false, in01);
T_LOAD_NHWC_WITH_DILATION({{SRC_DATA_TYPE}}, 1, 1, N0, {{SRC_TENSOR_TYPE}}, {{src}}, bout, yi1, xi0, g_ind_0, {{src}}_w, {{src}}_h, 1, 1, false, in10);
T_LOAD_NHWC_WITH_DILATION({{SRC_DATA_TYPE}}, 1, 1, N0, {{SRC_TENSOR_TYPE}}, {{src}}, bout, yi1, xi1, g_ind_0, {{src}}_w, {{src}}_h, 1, 1, false, in11);
)_";
if (is_data_type_float(_src->data_type()))
{
code += R"_(
const {{SRC_DATA_TYPE}} a = ({{SRC_DATA_TYPE}})(xi_f - (float)xi);
const {{SRC_DATA_TYPE}} b = ({{SRC_DATA_TYPE}})(1.f - a);
const {{SRC_DATA_TYPE}} a1 = ({{SRC_DATA_TYPE}})(yi_f - (float)yi);
const {{SRC_DATA_TYPE}} b1 = ({{SRC_DATA_TYPE}})(1.f - a1);
// Calculate the output
{{dst}}[0].v = ((in00[0].v * b * b1) + (in01[0].v * a * b1) + (in10[0].v * b * a1) + (in11[0].v * a * a1));
)_";
}
else
{
code += R"_(
const float a = (xi_f - (float)xi);
const float b = (1.f - a);
const float a1 = (yi_f - (float)yi);
const float b1 = (1.f - a1);
{{dst}}[0].v = CONVERT_SAT(
(CONVERT(in00[0].v, VEC_DATA_TYPE(float, N0)) * b * b1) +
(CONVERT(in01[0].v, VEC_DATA_TYPE(float, N0)) * a * b1) +
(CONVERT(in10[0].v, VEC_DATA_TYPE(float, N0)) * b * a1) +
(CONVERT(in11[0].v, VEC_DATA_TYPE(float, N0)) * a * a1), VEC_DATA_TYPE({{DST_DATA_TYPE}}, N0));
)_";
}
}
else
{
ARM_COMPUTE_ERROR("Unsupported interpolation policy");
}
code += R"_(
g_dst_indirect_y[0].v = g_ind_1 + (yo * (int)({{arg_dst}}_w)) + bout * (int)({{arg_dst}}_w * {{arg_dst}}_h);
}
//------------------ END KERNEL {{meta_kernel_id}} ---------------------
)_";
return code;
}
void ClTemplateResize::declare_variables(GpuKernelVariableTable &vtable,
const IGpuTemplateComponentWriter::ComponentGroup &comp_group) const
{
vtable.declare_variable(comp_group, _src, GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer),
"src");
vtable.declare_variable(comp_group, _dst, GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer),
"dst");
}
TagLUT ClTemplateResize::get_tag_lut(const GpuKernelVariableTable &vtable,
const IGpuTemplateComponentWriter::ComponentGroup &comp_group) const
{
TagLUT lut{};
// Arguments and global shared variables
lut["src"] = vtable.get_variable(_src);
lut["dst"] = vtable.get_variable(_dst);
const auto dst_argument = vtable.get_variable(comp_group.get_any_dst_tensor());
lut["arg_dst"] = dst_argument.uniq_name;
// Local build options
lut["meta_kernel_id"] = id();
lut["SRC_DATA_TYPE"] = get_cl_type_from_data_type(_src->data_type());
lut["SRC_TENSOR_TYPE"] = "BUFFER";
lut["DST_DATA_TYPE"] = get_cl_type_from_data_type(_dst->data_type());
lut["CONSTANT_VALUE"] = string_from_pixel_value(0, _src->data_type());
const float scale_x =
scale_utils::calculate_resize_ratio(_src->dimension(1), _dst->dimension(1), _attributes.align_corners());
const float scale_y =
scale_utils::calculate_resize_ratio(_src->dimension(2), _dst->dimension(2), _attributes.align_corners());
lut["SCALE_X"] = float_to_string_with_full_precision(scale_x);
lut["SCALE_Y"] = float_to_string_with_full_precision(scale_y);
return lut;
}
CLBuildOptions ClTemplateResize::get_build_options(const IGpuTemplateComponentWriter::ComponentGroup &comp_group) const
{
const Window root_window = comp_group.get_root_component()->template_writer()->get_window();
const unsigned int n0 = root_window.x().step();
const unsigned int m0 = root_window.y().step();
const unsigned int partial_n0 = _dst->dimension(0) % n0;
CLBuildOptions build_opts;
build_opts.add_option("-DN0=" + support::cpp11::to_string(n0));
build_opts.add_option("-DM0=" + support::cpp11::to_string(m0));
build_opts.add_option("-DPARTIAL_N0=" + support::cpp11::to_string(partial_n0));
return build_opts;
}
std::string ClTemplateResize::get_config_id() const
{
std::string config_id{};
config_id += "resize_";
config_id +=
(_attributes.interpolation_policy() == InterpolationPolicy::NEAREST_NEIGHBOR ? "NEAREST_NEIGHBOR" : "");
config_id += (_attributes.interpolation_policy() == InterpolationPolicy::BILINEAR ? "BILINEAR" : "");
config_id += "_";
config_id += (_attributes.sampling_policy() == SamplingPolicy::CENTER ? "center" : "topleft");
config_id += "_";
config_id += support::cpp11::to_string(_dst->dimension(0));
config_id += "_";
config_id += support::cpp11::to_string(_dst->dimension(1));
config_id += "_";
config_id += support::cpp11::to_string(_dst->dimension(2));
config_id += "_";
config_id += support::cpp11::to_string(_dst->dimension(3));
return config_id;
}
std::set<std::string> ClTemplateResize::get_headers_list() const
{
return std::set<std::string>{"helpers.h", "tile_helpers.h"};
}
Window ClTemplateResize::get_window() const
{
ARM_COMPUTE_ERROR_ON_MSG(_dst->tensor_shape().total_size() == 0U, "Destination tensor is not initialized");
const unsigned int n0 = adjust_vec_size(16 / _src->element_size(), _src->dimension(0));
Window win = calculate_max_window(*_dst, Steps(n0));
return win.collapse(win, Window::DimZ);
}
} // namespace dynamic_fusion
} // namespace experimental
} // namespace arm_compute