diff options
| author | Ramy Elgammal <ramy.elgammal@arm.com> | 2022-11-08 02:14:46 +0000 |
|---|---|---|
| committer | Ramy Elgammal <ramy.elgammal@arm.com> | 2022-11-29 10:24:59 +0000 |
| commit | 404462af4ca002ece819161a03a4bdb19a87abf2 (patch) | |
| tree | 09cf812530afcbe3fc524ce7eded5f06129e0889 /src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h | |
| parent | 03b2971ac69a86f10a1566938d1a25afee15746c (diff) | |
| download | ComputeLibrary-404462af4ca002ece819161a03a4bdb19a87abf2.tar.gz | |
Adding GpuAdd to dynamic fusion operators
- Provide support for Add operator
- Auto initialize the destination tensor before testing fusion in conv2d
and elementwise binary ops.
Resolves: COMPMID-5518
Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com>
Change-Id: Ibd815020f02b57f88eea7c2921bdcf98605d99c5
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8617
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h')
| -rw-r--r-- | src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h b/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h index c85ddf5a2c..328e942955 100644 --- a/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h +++ b/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h @@ -48,6 +48,9 @@ class IGpuTemplateComponentWriter public: using ComponentGroup = GpuKernelComponentGroup; + /**For now all kernel intermeditate/destination tensors are expected to be of type Tensor_4D_t_Buffer*/ + static constexpr GpuKernelArgumentInfo::Type common_tensor_type = GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer; + public: /** Constructor * |