diff options
| author | Gunes Bayir <gunes.bayir@arm.com> | 2022-03-10 21:21:01 +0000 |
|---|---|---|
| committer | Gunes Bayir <gunes.bayir@arm.com> | 2022-03-17 17:41:21 +0000 |
| commit | 8a87983c90299dfc7d6fbda3dba312e7603d7074 (patch) | |
| tree | ad1299cd902e6b2e5662f3f6e1b8fd12835b8469 /src/core/experimental/dynamic_fusion/ClKernelBuildingAPI.h | |
| parent | 193cad36d8ff70792562390b554304cc19284f61 (diff) | |
| download | ComputeLibrary-8a87983c90299dfc7d6fbda3dba312e7603d7074.tar.gz | |
Setup automatic kernel window for dynamic fusion
The window of the fused kernels is equal to the root kernel's window.
For this reason, the following kernels does not change the window and
intermediate tensors have the same shape and info as the destination tensor.
Resolves: COMPMID-5152
Change-Id: I25fe2fab8304ecaabfc2e4ade9bbf31a600a5033
Signed-off-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7316
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Diffstat (limited to 'src/core/experimental/dynamic_fusion/ClKernelBuildingAPI.h')
| -rw-r--r-- | src/core/experimental/dynamic_fusion/ClKernelBuildingAPI.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/core/experimental/dynamic_fusion/ClKernelBuildingAPI.h b/src/core/experimental/dynamic_fusion/ClKernelBuildingAPI.h index 15622c848d..27ab294cc9 100644 --- a/src/core/experimental/dynamic_fusion/ClKernelBuildingAPI.h +++ b/src/core/experimental/dynamic_fusion/ClKernelBuildingAPI.h @@ -100,13 +100,13 @@ struct ClKernelComponentDescriptor /** Component: Tensor Argument */ struct ClTensorDescriptor { - ClTensorDescriptor(const ITensorInfo *info, unsigned int dim) + ClTensorDescriptor(ITensorInfo *info, unsigned int dim) : tensor_info(info), slice_dim(dim) { } - const ITensorInfo *tensor_info; - unsigned int slice_dim; + ITensorInfo *tensor_info; + unsigned int slice_dim; }; Status add_tensor_argument(ClKernelBlueprint &, const ClTensorDescriptor &, ArgumentID &); @@ -133,8 +133,8 @@ struct GemmNativeDescriptor int32_t b_offset{}; }; -Status add_kcomp_gemm_native(ClKernelBlueprint &, const ClKernelComponentDescriptor &, const GemmNativeDescriptor &, ArgumentID input_id, - ArgumentID weights_id, ArgumentID bias_id, ArgumentID &dst_id); +Status add_kcomp_gemm_native(ClKernelBlueprint &, const ClKernelComponentDescriptor &, const GemmNativeDescriptor &, + ArgumentID input_id, ArgumentID weights_id, ArgumentID bias_id, ArgumentID &dst_id); /** Component: Eltwise Add */ struct EltwiseAddDescriptor |