diff options
author | ramelg01 <ramy.elgammal@arm.com> | 2022-02-08 09:38:17 +0000 |
---|---|---|
committer | Ramy Elgammal <ramy.elgammal@arm.com> | 2022-02-10 10:40:51 +0000 |
commit | bb6877ad4542943e718ac48727f238600fb8257c (patch) | |
tree | d4e4de9ebf9db10648a7fe772821610d2d74de8f /src/gpu/cl/kernels/ClWinogradOutputTransformKernel.h | |
parent | 6863fa061d34cb9d418872ff720c0e4ef4f8dbbb (diff) | |
download | ComputeLibrary-bb6877ad4542943e718ac48727f238600fb8257c.tar.gz |
Improve start-up time for winograd_output_transform_*_nhwc
- pass tensor's dimensions at runtime rather than compile time
- Add guard macro to compile only kernel of internest
Resolves: COMPMID-5120
Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com>
Change-Id: I87c3b56ce0cd3c97ffdeabdd9c5d433f361bb005
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7101
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/gpu/cl/kernels/ClWinogradOutputTransformKernel.h')
-rw-r--r-- | src/gpu/cl/kernels/ClWinogradOutputTransformKernel.h | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/gpu/cl/kernels/ClWinogradOutputTransformKernel.h b/src/gpu/cl/kernels/ClWinogradOutputTransformKernel.h index 674d52c904..6f018967d0 100644 --- a/src/gpu/cl/kernels/ClWinogradOutputTransformKernel.h +++ b/src/gpu/cl/kernels/ClWinogradOutputTransformKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2021 Arm Limited. + * Copyright (c) 2018-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -77,7 +77,11 @@ public: private: using WinogradKey = std::pair<std::pair<int, int>, std::pair<int, int>>; - bool _is_nhwc{ false }; + bool _is_nhwc{ false }; + int32_t _src_height{ 0 }; + int32_t _dst_width{ 0 }; + int32_t _dst_height{ 0 }; + int32_t _num_tiles_x{ 0 }; }; } // namespace kernels } // namespace opencl |