diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2018-04-16 11:48:56 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:19 +0000 |
| commit | 95a3f55bbfb954aee58b2a13276035aa7b9e36a2 (patch) | |
| tree | 19d1b40d23e0a3a86c6f3f03c0dc1a3da93f46d0 /src/runtime/NEON/functions/NEConvolutionLayer.cpp | |
| parent | 5f0124db5cb7de18930b474af7cb722987e088fa (diff) | |
| download | ComputeLibrary-95a3f55bbfb954aee58b2a13276035aa7b9e36a2.tar.gz | |
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 <pablo.tello@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEConvolutionLayer.cpp')
| -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; } |