diff options
author | SiCong Li <sicong.li@arm.com> | 2020-08-26 13:55:15 +0100 |
---|---|---|
committer | SiCong Li <sicong.li@arm.com> | 2020-08-28 13:00:32 +0000 |
commit | 5bdde8509542e942e908e9d508dd39c73194abfb (patch) | |
tree | 23ebe231e1bd04f0923589878f6c7cf930b033cf /arm_compute | |
parent | ce7c44117918ed4d01c60010565d942d2221748e (diff) | |
download | ComputeLibrary-5bdde8509542e942e908e9d508dd39c73194abfb.tar.gz |
COMPMID-3670 Extend cl image support to f16 in CLGEMMReshapeRHSMatrixKernel
Change-Id: Ic0569fe9ed99e61084b601ce84ddc7ef288d1899
Signed-off-by: SiCong Li <sicong.li@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3852
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/CL/kernels/CLGEMMReshapeRHSMatrixKernel.h | 6 |
1 files changed, 3 insertions, 3 deletions
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 |