openmp.patch


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

diff --git a/src/tensors/cpu/tensor_operators.cpp b/src/tensors/cpu/tensor_operators.cpp
index ed45e91..a375494 100755
--- a/src/tensors/cpu/tensor_operators.cpp
+++ b/src/tensors/cpu/tensor_operators.cpp
@@ -951,13 +951,13 @@ void CrossEntropyPick(Tensor out, Tensor in, Tensor labelIndices, float labelSmo
     }
 
     float sumexp = 0.f;
-    #pragma omp simd reduction(+ : sum)
+    #pragma omp simd reduction(+ : sumexp)
     for(int i = 0; i < cols; ++i) {
       sumexp += std::exp(sp[i] - max);
     }
 
     float mean = 0.f;
-    #pragma omp simd reduction(+ : sum)
+    #pragma omp simd reduction(+ : mean)
     for(int i = 0; i < cols; ++i) {
       mean += sp[i] - max;
     }