diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-08-17 16:30:12 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-08-18 12:42:16 +0000 |
commit | df5bcb67e895610c0a7360753817fea13faabd4e (patch) | |
tree | bbc28cc2a2029071265ca47f0bcad6ffe0f30d93 /src/runtime/cpu/operators/CpuConv2d.cpp | |
parent | ebbd5291bca6493229ef276aa67d5ec533141277 (diff) | |
download | ComputeLibrary-df5bcb67e895610c0a7360753817fea13faabd4e.tar.gz |
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>
Diffstat (limited to 'src/runtime/cpu/operators/CpuConv2d.cpp')
-rw-r--r-- | src/runtime/cpu/operators/CpuConv2d.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/cpu/operators/CpuConv2d.cpp b/src/runtime/cpu/operators/CpuConv2d.cpp index ec7833324a..809663a918 100644 --- a/src/runtime/cpu/operators/CpuConv2d.cpp +++ b/src/runtime/cpu/operators/CpuConv2d.cpp @@ -63,7 +63,7 @@ void CpuConv2d::configure(ITensorInfo *input, ITensorInfo *weights, const ITenso 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 @@ Status CpuConv2d::validate(const ITensorInfo *input, const ITensorInfo *weights, 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)); |