From 883f489da93e88d74aa0dfb206c56697ba0e63f0 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 3 Aug 2018 13:41:33 +0100 Subject: COMPMID-1475: (OCLGrind) FP exception in NEGEMMConvolution Change-Id: I986099c269498cc7971b10ee634dba721954546e Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/142647 Tested-by: Jenkins Reviewed-by: Gian Marco Iodice --- src/runtime/CL/functions/CLGEMMConvolutionLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/runtime') diff --git a/src/runtime/CL/functions/CLGEMMConvolutionLayer.cpp b/src/runtime/CL/functions/CLGEMMConvolutionLayer.cpp index fde15613d6..d58f82e9c6 100644 --- a/src/runtime/CL/functions/CLGEMMConvolutionLayer.cpp +++ b/src/runtime/CL/functions/CLGEMMConvolutionLayer.cpp @@ -99,7 +99,7 @@ CLGEMMConvolutionLayer::CLGEMMConvolutionLayer(std::shared_ptr m void CLGEMMConvolutionLayer::configure_mm(const ICLTensor *input, const ICLTensor *weights, ICLTensor *output, int gemm_3d_depth) { ARM_COMPUTE_ERROR_ON_NULLPTR(input, weights); - ARM_COMPUTE_ERROR_THROW_ON(validate_mm(input->info(), weights->info(), output->info(), _skip_im2col)); + ARM_COMPUTE_ERROR_THROW_ON(validate_mm(input->info(), weights->info(), output->info(), gemm_3d_depth, _skip_im2col)); if(_is_quantized) { -- cgit v1.2.1