From 3ba2d55f7e2554b401cb191531d1be87eb07d76a Mon Sep 17 00:00:00 2001
From: chenjiaoAngel <chenjiaobuaa@126.com>
Date: Thu, 7 Nov 2019 14:48:53 +0800
Subject: [PATCH] add openmp, test=develop

---
 lite/backends/arm/math/pooling.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lite/backends/arm/math/pooling.cc b/lite/backends/arm/math/pooling.cc
index d023ed35fd..a857e9830c 100644
--- a/lite/backends/arm/math/pooling.cc
+++ b/lite/backends/arm/math/pooling.cc
@@ -86,6 +86,7 @@ void pooling_basic(const float* din,
     }
   } else {
     for (int ind_n = 0; ind_n < num; ++ind_n) {
+#pragma omp parallel for
       for (int ind_c = 0; ind_c < chin; ++ind_c) {
         for (int ind_h = 0; ind_h < hout; ++ind_h) {
           int sh = ind_h * stride_h;
-- 
GitLab