diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-23 17:47:53 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-23 18:05:53 +0000 |
| commit | a8297fb7433e1acfb7bb37703ab93b7ec4c39335 (patch) | |
| tree | 0cded55840c4d7e7851af737142ac7284ab57e09 | |
| parent | 6575ef8fa14324e718d1c111859a24725835a648 (diff) | |
| download | ComputeLibrary-a8297fb7433e1acfb7bb37703ab93b7ec4c39335.tar.gz | |
Pass fast math flag for correct GEMM3D validation support
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ic96bac64680e052e39c21d2e0eb9e618bcf259b4
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5989
Tested-by: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/runtime/cpu/operators/CpuGemmConvolution.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/cpu/operators/CpuGemmConvolution.cpp b/src/runtime/cpu/operators/CpuGemmConvolution.cpp index 7defc13b20..e7fae9da7b 100644 --- a/src/runtime/cpu/operators/CpuGemmConvolution.cpp +++ b/src/runtime/cpu/operators/CpuGemmConvolution.cpp @@ -208,7 +208,7 @@ Status CpuGemmConvolution::validate_gemm3d(const ITensorInfo *input_info, const const TensorInfo dummy_weights_info(TensorShape(4U, 4U), 1, data_type, weights_info->quantization_info()); const TensorInfo dummy_output_info(TensorShape(4U, 4U, gemm_3d_depth), 1, data_type, input_info->quantization_info()); - return validate_mm(&dummy_input_info, &dummy_weights_info, nullptr, &dummy_output_info, act_info, gemm_3d_depth, skip_im2col); + return validate_mm(&dummy_input_info, &dummy_weights_info, nullptr, &dummy_output_info, act_info, false, gemm_3d_depth, skip_im2col); } void CpuGemmConvolution::configure(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, ITensorInfo *dst, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, |