diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2023-07-07 11:25:57 +0100 |
|---|---|---|
| committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-07-20 08:48:15 +0000 |
| commit | ebfdb5a1ea73c2269eec5af492970c2174ab7d0f (patch) | |
| tree | 5c60d083100118f0a40e629dfa69b7a7373dd7fd /compute_kernel_writer/src/TensorUtils.h | |
| parent | 29dc9fc1d3d6e90746ba1173e3318b774dcf7bed (diff) | |
| download | ComputeLibrary-ebfdb5a1ea73c2269eec5af492970c2174ab7d0f.tar.gz | |
Integrate CLTensorArgument
- Add CLTensorArgument to query the components and storages as OpenCL
variables (or by values when possible)
- Add caching mechanism in CLTensorArgument to keep track of the components and storages used
- Add unit tests
Resolves COMPMID-5787
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: Ib39e1f77b097e5b907a296fe6b0d41bb4bcd4ffc
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9908
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Diffstat (limited to 'compute_kernel_writer/src/TensorUtils.h')
| -rw-r--r-- | compute_kernel_writer/src/TensorUtils.h | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/compute_kernel_writer/src/TensorUtils.h b/compute_kernel_writer/src/TensorUtils.h index 84eca084bb..bb0af5c0b9 100644 --- a/compute_kernel_writer/src/TensorUtils.h +++ b/compute_kernel_writer/src/TensorUtils.h @@ -22,8 +22,8 @@ * SOFTWARE. */ -#ifndef COMPUTE_KERNEL_WRITER_SRC_TENSORUTILS_H -#define COMPUTE_KERNEL_WRITER_SRC_TENSORUTILS_H +#ifndef CKW_SRC_TENSORUTILS_H +#define CKW_SRC_TENSORUTILS_H #include <cstdint> @@ -33,7 +33,7 @@ namespace ckw // Forward declarations enum class TensorDataLayout; enum class TensorDataLayoutComponent; -enum class TensorComponent : uint32_t; +enum class TensorComponentType : uint32_t; /** Get tensor dimension from a given data layout and data layout component * @@ -42,7 +42,7 @@ enum class TensorComponent : uint32_t; * * @return the @ref TensorComponent */ -TensorComponent get_tensor_dimension(TensorDataLayout layout, TensorDataLayoutComponent component); +TensorComponentType get_tensor_dimension(TensorDataLayout layout, TensorDataLayoutComponent component); /** Get tensor stride from a given data layout and data layout component * @@ -51,6 +51,7 @@ TensorComponent get_tensor_dimension(TensorDataLayout layout, TensorDataLayoutCo * * @return the @ref TensorComponent */ -TensorComponent get_tensor_stride(TensorDataLayout layout, TensorDataLayoutComponent component); +TensorComponentType get_tensor_stride(TensorDataLayout layout, TensorDataLayoutComponent component); } // namespace ckw -#endif /* COMPUTE_KERNEL_WRITER_SRC_TENSORUTILS_H */ + +#endif // CKW_SRC_TENSORUTILS_H |