From 95a3f55bbfb954aee58b2a13276035aa7b9e36a2 Mon Sep 17 00:00:00 2001 From: Anthony Barbier Date: Mon, 16 Apr 2018 11:48:56 +0100 Subject: COMPMID-959: Disabled winograd on Arm Cortex-A53 while we're working on a fix Change-Id: I29e6288ceb60002208bb9c9bb1a517e56e70c20e Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/127807 Reviewed-by: Pablo Tello Tested-by: Jenkins --- src/runtime/NEON/functions/NEConvolutionLayer.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/runtime/NEON/functions/NEConvolutionLayer.cpp') diff --git a/src/runtime/NEON/functions/NEConvolutionLayer.cpp b/src/runtime/NEON/functions/NEConvolutionLayer.cpp index f248821de6..afc354533d 100644 --- a/src/runtime/NEON/functions/NEConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEConvolutionLayer.cpp @@ -114,7 +114,11 @@ ConvolutionMethod NEConvolutionLayer::get_convolution_method(const ITensorInfo * if((input->data_type() == DataType::F32) && (weights->dimension(0) == 3) && (weights->dimension(1) == 3) && (weights->num_dimensions() <= 4) && (conv_info.stride().first == 1) && (conv_info.stride().second == 1) && (dilation == Size2D(1U, 1U))) { - return ConvolutionMethod::WINOGRAD; + //FIXME Until COMPMID-1041 is implemented Winograd is slower than GEMM on A53. + if(Scheduler::get().cpu_info().get_cpu_model() != CPUModel::A53) + { + return ConvolutionMethod::WINOGRAD; + } } return ConvolutionMethod::GEMM; } -- cgit v1.2.1