From 9f504d6bf49b434c0c40b071c0e0501c4927a657 Mon Sep 17 00:00:00 2001
From: Giorgio Arena <giorgio.arena@arm.com>
Date: Mon, 21 Feb 2022 11:11:29 +0000
Subject: 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>
---
 src/cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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;
-- 
cgit v1.2.1