diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-11-10 11:18:50 +0000 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-11-26 11:03:07 +0000 |
commit | 4fb5670e579f9c0e1403a8fe179a11a2c9f3a02f (patch) | |
tree | dabcf464ad75ee0f7437720f0131201d207624c8 /src/core/CL/cl_kernels/tile_helpers.h | |
parent | 1d4a320b7c638936a8b4a5c55ac59c61f63791a1 (diff) | |
download | ComputeLibrary-4fb5670e579f9c0e1403a8fe179a11a2c9f3a02f.tar.gz |
Rework gemm_reshape_rhs_(nt,t) with new macros
Resolves COMPMID-4891
Change-Id: Ifdf2a0eaed23347a1b4465ea8d58c11b72083952
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6741
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Diffstat (limited to 'src/core/CL/cl_kernels/tile_helpers.h')
-rw-r--r-- | src/core/CL/cl_kernels/tile_helpers.h | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/src/core/CL/cl_kernels/tile_helpers.h b/src/core/CL/cl_kernels/tile_helpers.h index cc20616867..30c37da1d0 100644 --- a/src/core/CL/cl_kernels/tile_helpers.h +++ b/src/core/CL/cl_kernels/tile_helpers.h @@ -130,6 +130,28 @@ #define TENSOR4D_T_STR(name, type) TENSOR4D_T_##type(name) #define TENSOR4D_T(name, type) TENSOR4D_T_STR(name, type) +#define TENSOR3D_T_IMAGE(name) \ + __read_only image2d_t name##_img, \ + __global uchar *name##_ptr, \ + uint name##_stride_y, \ + uint name##_stride_z, \ + uint name##_w, \ + uint name##_h, \ + uint name##_n, \ + uint name##_offset_first_element_in_bytes + +#define TENSOR3D_T_BUFFER(name) \ + __global uchar *name##_ptr, \ + uint name##_stride_y, \ + uint name##_stride_z, \ + uint name##_w, \ + uint name##_h, \ + uint name##_n, \ + uint name##_offset_first_element_in_bytes + +#define TENSOR3D_T_STR(name, type) TENSOR3D_T_##type(name) +#define TENSOR3D_T(name, type) TENSOR3D_T_STR(name, type) + #if !defined(UNROLL_WITH_PRAGMA) #define UNROLL_INCR(idx, step, macro) idx += (step); (macro) |