From 5bdde8509542e942e908e9d508dd39c73194abfb Mon Sep 17 00:00:00 2001 From: SiCong Li Date: Wed, 26 Aug 2020 13:55:15 +0100 Subject: COMPMID-3670 Extend cl image support to f16 in CLGEMMReshapeRHSMatrixKernel Change-Id: Ic0569fe9ed99e61084b601ce84ddc7ef288d1899 Signed-off-by: SiCong Li Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3852 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h') diff --git a/arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h b/arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h index 5f953ddf8d..557f71b07d 100644 --- a/arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h +++ b/arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h @@ -53,7 +53,7 @@ public: * Since the OpenCL image object is created importing the OpenCL buffer, the following conditions are required: * -# rhs_info.n0 can only be 4, 8 and 16 * -# rhs_info.k0 can only be 4, 8 and 16 - * -# Data type can only be F32 + * -# Data type can only be F32, F16 * -# The platform should support the OpenCL cl_khr_image2d_from_buffer extension * -# output width should be less or equal to (CL_DEVICE_IMAGE2D_MAX_WIDTH * 4) * -# output (height * depth) should be less or equal to CL_DEVICE_IMAGE2D_MAX_HEIGHT @@ -77,7 +77,7 @@ public: * Since the OpenCL image object is created importing the OpenCL buffer, the following conditions are required: * -# rhs_info.n0 can only be 4, 8 and 16 * -# rhs_info.k0 can only be 4, 8 and 16 - * -# Data type can only be F32 + * -# Data type can only be F32, F16 * -# The platform should support the OpenCL cl_khr_image2d_from_buffer extension * -# output width should be less or equal to (CL_DEVICE_IMAGE2D_MAX_WIDTH * 4) * -# output (height * depth) should be less or equal to CL_DEVICE_IMAGE2D_MAX_HEIGHT @@ -102,7 +102,7 @@ public: * Since the OpenCL image object is created importing the OpenCL buffer, the following conditions are required: * -# rhs_info.n0 can only be 4, 8 and 16 * -# rhs_info.k0 can only be 4, 8 and 16 - * -# Data type can only be F32 + * -# Data type can only be F32, F16 * -# The platform should support the OpenCL cl_khr_image2d_from_buffer extension * -# output width should be less or equal to (CL_DEVICE_IMAGE2D_MAX_WIDTH * 4) * -# output (height * depth) should be less or equal to CL_DEVICE_IMAGE2D_MAX_HEIGHT -- cgit v1.2.1