diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/runtime/NEON/functions/NEConvolutionLayer.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
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; } |