提交 c23719f2 编写于 作者: M Michal W. Tarnowski 提交者: Liangliang He

Added missing pragmas (#312)

上级 a1dcdcb5
...@@ -433,6 +433,7 @@ inline void TensorEltwise(const EltwiseType type, ...@@ -433,6 +433,7 @@ inline void TensorEltwise(const EltwiseType type,
output[i] = std::pow(input0[i], input1[i]); output[i] = std::pow(input0[i], input1[i]);
} }
} else { } else {
#pragma omp parallel for schedule(runtime)
for (index_t i = 0; i < size; ++i) { for (index_t i = 0; i < size; ++i) {
output[i] = std::pow(input1[i], input0[i]); output[i] = std::pow(input1[i], input0[i]);
} }
...@@ -552,6 +553,7 @@ inline void TensorScalarEltwise(const EltwiseType type, ...@@ -552,6 +553,7 @@ inline void TensorScalarEltwise(const EltwiseType type,
output[i] = std::pow(input0[i], input1); output[i] = std::pow(input0[i], input1);
} }
} else { } else {
#pragma omp parallel for schedule(runtime)
for (index_t i = 0; i < size; ++i) { for (index_t i = 0; i < size; ++i) {
output[i] = std::pow(input1, input0[i]); output[i] = std::pow(input1, input0[i]);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册