Merge branch 'reduction' into 'master'

Fix openmp bug: See merge request !811

Merge branch 'reduction' into 'master'
Fix openmp bug: See merge request !811
cc0d66e3 · 刘托 · 68518b94 · 949c1bd1 · cc0d66e3
显示空白变更内容
内联并排

Showing with 0 addition and 1 deletion

mace/kernels/reduce_mean.h mace/kernels/reduce_mean.h +0 -1

未找到文件。
--- a/mace/kernels/reduce_mean.h
+++ b/mace/kernels/reduce_mean.h
@@ -110,7 +110,6 @@ struct ReduceMeanFunctor : ReduceFunctorBase {
      case 1:
        if (reduce_first_axis_) {
          T sum = 0;
-#pragma omp parallel for reduction(+:sum)
          for (int i = 0; i < data_reshape_[0]; ++i) {
            sum = sum + input_ptr[i];
          }