commit a7a16943ab603e8d5697b872421a91151a57cc7d
author Georgios Pinitas <georgios.pinitas@arm.com> Tue Aug 17 16:30:12 2021 +0100
committer Freddie Liardet <frederick.liardet@arm.com> Wed Aug 18 13:44:53 2021 +0100
tree bbc28cc2a2029071265ca47f0bcad6ffe0f30d93
parent e28942a05a0903f43344f3234c1f6e79fe21973d

Enable fast_math on CpuGemmConvolution

Fast-math mode enables faster GEMM implementations to be used assuming
an accuracy impact is acceptable.

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I0d0f23a2843426cc37a31f63022c790249b71c69
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6122
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

src/runtime/cpu/operators/CpuConv2d.cpp

diff --git a/src/runtime/cpu/operators/CpuConv2d.cpp b/src/runtime/cpu/operators/CpuConv2d.cpp
index ec78333..809663a 100644
--- a/src/runtime/cpu/operators/CpuConv2d.cpp
+++ b/src/runtime/cpu/operators/CpuConv2d.cpp
@@ -63,7 +63,7 @@
         case ConvolutionMethod::GEMM:
         {
             auto f = std::make_unique<CpuGemmConvolution>();
-            f->configure(input, weights, biases, output, conv_info, weights_info, dilation, act_info);
+            f->configure(input, weights, biases, output, conv_info, weights_info, dilation, act_info, enable_fast_math);
             _function = std::move(f);
             break;
         }
@@ -101,7 +101,7 @@
             ARM_COMPUTE_RETURN_ON_ERROR(CpuWinogradConv2d::validate(input, weights, biases, output, conv_info, act_info, enable_fast_math));
             break;
         case ConvolutionMethod::GEMM:
-            ARM_COMPUTE_RETURN_ON_ERROR(CpuGemmConvolution::validate(input, weights, biases, output, conv_info, weights_info, dilation, act_info));
+            ARM_COMPUTE_RETURN_ON_ERROR(CpuGemmConvolution::validate(input, weights, biases, output, conv_info, weights_info, dilation, act_info, enable_fast_math));
             break;
         case ConvolutionMethod::GEMM_CONV2D:
             ARM_COMPUTE_RETURN_ON_ERROR(CpuGemmDirectConv2d::validate(input, weights, biases, output, info));