From b87b95e13cce28523e25d6a50af3147edfb43357 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Mon, 21 Jan 2019 17:14:31 +0000 Subject: COMPMID-1899: Fix NaN issue in CLGEMMMatrixMultiplyReshapedKernel Change-Id: Ide950b46c4d41de230c272c7044a03f4f9f237ed Reviewed-on: https://review.mlplatform.org/548 Reviewed-by: Georgios Pinitas Reviewed-by: Giuseppe Rossini Tested-by: Arm Jenkins --- src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp') diff --git a/src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp b/src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp index ca0ebd63df..242924b1d1 100644 --- a/src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp +++ b/src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp @@ -133,6 +133,7 @@ void CLGEMMReshapeLHSMatrixKernel::configure(const ICLTensor *input, ICLTensor * build_opts.add_option("-DM0=" + support::cpp11::to_string(lhs_info.m0)); build_opts.add_option("-DK0=" + support::cpp11::to_string(lhs_info.k0)); build_opts.add_option("-DV0=" + support::cpp11::to_string(lhs_info.v0)); + build_opts.add_option("-DSRC_WIDTH=" + support::cpp11::to_string(input->info()->dimension(0))); build_opts.add_option_if(lhs_info.interleave, "-DINTERLEAVE"); build_opts.add_option_if(_reinterpret_input_as_3d, "-DREINTERPRET_INPUT_AS_3D"); build_opts.add_option_if(_reinterpret_input_as_3d, "-DHEIGHT_GEMM3D=" + support::cpp11::to_string(input->info()->dimension(1))); -- cgit v1.2.1