From c23719f2061ee04637d5afc09235c8bbf44b4906 Mon Sep 17 00:00:00 2001
From: "Michal W. Tarnowski" <michal.w.tarnowski@hotmail.com>
Date: Tue, 15 Jan 2019 03:11:31 +0100
Subject: [PATCH] Added missing pragmas (#312)

---
 mace/ops/eltwise.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mace/ops/eltwise.cc b/mace/ops/eltwise.cc
index 4166f4d1..8d1cae4b 100644
--- a/mace/ops/eltwise.cc
+++ b/mace/ops/eltwise.cc
@@ -433,6 +433,7 @@ inline void TensorEltwise(const EltwiseType type,
           output[i] = std::pow(input0[i], input1[i]);
         }
       } else {
+#pragma omp parallel for schedule(runtime)
         for (index_t i = 0; i < size; ++i) {
           output[i] = std::pow(input1[i], input0[i]);
         }
@@ -552,6 +553,7 @@ inline void TensorScalarEltwise(const EltwiseType type,
           output[i] = std::pow(input0[i], input1);
         }
       } else {
+#pragma omp parallel for schedule(runtime)
         for (index_t i = 0; i < size; ++i) {
           output[i] = std::pow(input1, input0[i]);
         }
-- 
GitLab