diff options
author | Ramy Elgammal <ramy.elgammal@arm.com> | 2022-12-14 15:39:29 +0000 |
---|---|---|
committer | Ramy Elgammal <ramy.elgammal@arm.com> | 2023-01-06 15:23:07 +0000 |
commit | f800adf185e966b16385f65b9c7250766949dbe4 (patch) | |
tree | bd857f518c725fc62e49dedce187b5188c08e1f4 /src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateCast.cpp | |
parent | 3a1e1256946028cb13f31521caec9f08235c7332 (diff) | |
download | ComputeLibrary-f800adf185e966b16385f65b9c7250766949dbe4.tar.gz |
Implement dynamic fusion reshape operator
Resolves: COMPMID-5522
Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com>
Change-Id: If4e5736a2f7ff42e70276d7f4e0f3ebcb38414e6
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8881
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateCast.cpp')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateCast.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateCast.cpp b/src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateCast.cpp index dcb43f9783..e850fee313 100644 --- a/src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateCast.cpp +++ b/src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateCast.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2022 Arm Limited. + * Copyright (c) 2022-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -54,7 +54,7 @@ std::string ClTemplateCast::get_component_code(const ComponentGroup &comp_group) ARM_COMPUTE_UNUSED(comp_group); const std::string kernel_name = get_name(); - const auto is_root = (comp_group.get_root_component()->id() == this->id()); + const auto is_root = (comp_group.get_root_component()->id() == this->id()); std::string code = R"_( //------------------ START KERNEL {{meta_kernel_id}} CAST --------------------- @@ -204,7 +204,7 @@ Window ClTemplateCast::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)); + const unsigned int n0 = adjust_vec_size(16 / _dst->element_size(), _dst->dimension(0)); Window win = calculate_max_window(*_dst, Steps(n0)); return win.collapse(win, Window::DimZ); } |