diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-01-21 17:14:31 +0000 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-01-21 18:52:34 +0000 |
| commit | b87b95e13cce28523e25d6a50af3147edfb43357 (patch) | |
| tree | 7261a482a140c4fe082caa1c172ead302be23240 /src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp | |
| parent | a0a0e29f635de08092c2325f8f049ffb286aabaf (diff) | |
| download | ComputeLibrary-b87b95e13cce28523e25d6a50af3147edfb43357.tar.gz | |
COMPMID-1899: Fix NaN issue in CLGEMMMatrixMultiplyReshapedKernel
Change-Id: Ide950b46c4d41de230c272c7044a03f4f9f237ed
Reviewed-on: https://review.mlplatform.org/548
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLGEMMReshapeLHSMatrixKernel.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
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))); |