Fix OMP compilation (#824)

* Fix omp variable names

1 files changed, 2 insertions, 2 deletions

diff --git a/src/tensors/cpu/tensor_operators.cpp b/src/tensors/cpu/tensor_operators.cpp
index 3eded6b2..1191a2be 100755
--- a/src/tensors/cpu/tensor_operators.cpp
+++ b/src/tensors/cpu/tensor_operators.cpp
@@ -914,13 +914,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;
     }