From 3c1d9817b5e4303eab5aebed513928611bde7366 Mon Sep 17 00:00:00 2001
From: HappyAngel <chenjiaobuaa@126.com>
Date: Sun, 7 Jun 2020 22:39:17 -0500
Subject: [PATCH] [arm] fix 3x3s2p0 avg pooling crash (#3754)

* fix pooling overflow, test=develop

* fix conflict test=develop
---
 lite/backends/arm/math/pooling.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lite/backends/arm/math/pooling.cc b/lite/backends/arm/math/pooling.cc
index aaa9b9ff4c..3e6cbff066 100644
--- a/lite/backends/arm/math/pooling.cc
+++ b/lite/backends/arm/math/pooling.cc
@@ -2701,7 +2701,9 @@ void pooling3x3s2p0_avg(const float* din,
     w_unroll_size -= 1;
     w_unroll_remian = wout - w_unroll_size * 4;
   }
-
+  //  do overflow process
+  w_unroll_size -= 1;
+  w_unroll_remian += 4;
   auto zero_ptr =
       static_cast<float*>(TargetMalloc(TARGET(kARM), win * sizeof(float)));
   memset(zero_ptr, 0, win * sizeof(float));
-- 
GitLab