Fix CpuPool2d regression on A53/A55 due to mws

Resolves COMPMID-5045

Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: I85562a96cc6998567a96b6f43d1a0145f7c7b330
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7177
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

1 files changed, 5 insertions, 1 deletions

diff --git a/src/cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp b/src/cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp
index a8c6a5e611..aad1daaf53 100644
--- a/src/cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp
+++ b/src/cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021 Arm Limited.
+ * Copyright (c) 2021-2022 Arm Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -285,6 +285,10 @@ size_t CpuPool2dAssemblyWrapperKernel::get_mws(const CPUInfo &platform, size_t t
     {
         return 10240;
     }
+    else if(platform.get_cpu_model() == CPUModel::A53 || platform.get_cpu_model() == CPUModel::A55r0 || platform.get_cpu_model() == CPUModel::A55r1)
+    {
+        return 1;
+    }
     else
     {
         return 9216;