Skip to content

Commit d14858e

Browse files
committed
MKLDNN elementwise_mul: Parallelize mul
1 parent ed31936 commit d14858e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

paddle/fluid/operators/elementwise_mul_mkldnn_op.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ class ElementwiseMulMKLDNNKernel : public framework::OpKernel<T> {
144144

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

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

0 commit comments

Comments
 (0)