diff options
author | SiCongLi <sicong.li@arm.com> | 2021-03-04 10:27:03 +0000 |
---|---|---|
committer | SiCong Li <sicong.li@arm.com> | 2021-03-08 14:25:56 +0000 |
commit | eda87d40532304482acade655580930329a0bb8b (patch) | |
tree | d17760869eb047dc8adc600ebf870f5fee25717c | |
parent | 71c3cde26eaf973ad6c7d11ede60e2f864835c67 (diff) | |
download | ComputeLibrary-eda87d40532304482acade655580930329a0bb8b.tar.gz |
Fix cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint example
CLGEMMLowpReshapedOnlyRHSKernel only accepts 1D bias.
Resolves COMPMID-4295
Change-Id: Ib41c42ded3a6e6e7009a238b2deeffb90aa9ea41
Signed-off-by: SiCongLi <sicong.li@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5218
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r-- | examples/gemm_tuner/cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/gemm_tuner/cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp b/examples/gemm_tuner/cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp index 95431ed50c..d8f8f1498a 100644 --- a/examples/gemm_tuner/cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp +++ b/examples/gemm_tuner/cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp @@ -193,7 +193,7 @@ public: lhs.allocator()->init(TensorInfo(TensorShape(params.K, params.M, params.B), 1, params.data_type)); rhs.allocator()->init(TensorInfo(TensorShape(params.N, params.K, params.B), 1, params.data_type)); - bias.allocator()->init(TensorInfo(TensorShape(params.N, 1, params.B), 1, DataType::S32)); + bias.allocator()->init(TensorInfo(TensorShape(params.N), 1, DataType::S32)); dst.allocator()->init(TensorInfo(TensorShape(params.N, params.M, params.B), 1, params.data_type)); // Set arbitrary quantization information (non-zero offset to ensure offset contribution stage is included) |