MKLDNN elementwise_mul: Parallelize mul

6 years ago · d14858e4ba
parent ed31936ba1
commit d14858e4ba
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/operators/elementwise_mul_mkldnn_op.cc
+++ b/paddle/fluid/operators/elementwise_mul_mkldnn_op.cc
@ -144,6 +144,7 @@ class ElementwiseMulMKLDNNKernel : public framework::OpKernel<T> {

          mul_func_t mul_func = (mul_func_t) mul.getCode();

+          #pragma omp parallel for collapse(2)
          for (int ni = 0; ni < n; ni++) {
            for (int ci = 0; ci < C; ci++) {
              auto ptr_x =