diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-07-11 15:54:56 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:10 +0000 |
commit | e8bd2c729546e59aa0adc241976ea91fc6f25b52 (patch) | |
tree | deda6658ad22ee6f68f96f6221caed5aa80acd10 /src/core/CL/kernels/CLGEMMMatrixMultiplyKernel.cpp | |
parent | ed32f43174ce45cafe9d93e1a0b92cbebbcb7493 (diff) | |
download | ComputeLibrary-e8bd2c729546e59aa0adc241976ea91fc6f25b52.tar.gz |
COMPMID-1384: graph_mobilenet fails for NHWC on OpenCL
Makes GEMM3D account top padding when jumping among planes.
Change-Id: Ia7c16cfa5498de106774ce42cbc4716e9f43195b
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/139612
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLGEMMMatrixMultiplyKernel.cpp')
-rw-r--r-- | src/core/CL/kernels/CLGEMMMatrixMultiplyKernel.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/core/CL/kernels/CLGEMMMatrixMultiplyKernel.cpp b/src/core/CL/kernels/CLGEMMMatrixMultiplyKernel.cpp index 814cbb631f..0c629af788 100644 --- a/src/core/CL/kernels/CLGEMMMatrixMultiplyKernel.cpp +++ b/src/core/CL/kernels/CLGEMMMatrixMultiplyKernel.cpp @@ -358,8 +358,9 @@ void CLGEMMMatrixMultiplyKernel::run(const Window &window, cl::CommandQueue &que if(_is_gemm3d) { // Pass bottom paddings to the kernel if the output has to be reinterpreted as 3D tensor - const unsigned int idx0 = 3 * num_arguments_per_2D_tensor() + 3; - _kernel.setArg<cl_uint>(idx0, static_cast<unsigned int>(_output->info()->padding().bottom)); + const unsigned int idx0 = 3 * num_arguments_per_2D_tensor() + 3; + const unsigned int total_cross_plane_pad = _output->info()->padding().top + _output->info()->padding().bottom; + _kernel.setArg<cl_uint>(idx0, static_cast<unsigned int>(total_cross_plane_pad)); } do |