1 files changed, 5 insertions, 1 deletions
diff --git a/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp b/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp
index 934e38b054..6c235df59f 100644
--- a/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp
+++ b/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp
@@ -363,10 +363,14 @@ size_t CpuDepthwiseConv2dAssemblyWrapperKernel::get_mws(const CPUInfo &platform,
     {
         return 10240;
     }
-    else 
+    else if (platform.get_cpu_model() == CPUModel::A76)
     {
         return 9216;
     }
+    else
+    {
+        return ICPPKernel::default_mws;
+    }
 }
 } // namespace kernels
 } // namespace cpu