Skip to content

Commit 742300b

Browse files
committed
fix unkown omp pragmas
1 parent b9dbb7c commit 742300b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/fluid/operators/fc_op.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,9 @@ class FCOpKernel : public framework::OpKernel<T> {
145145

146146
if (bias) {
147147
const T* bias_data = bias->data<T>();
148+
#ifdef PADDLE_WITH_MKLML
148149
#pragma omp parallel for if (FLAGS_paddle_num_threads > 1)
150+
#endif
149151
for (int bs = 0; bs < in_dims[0]; bs++) {
150152
blas.AXPY(w_dims[1], static_cast<T>(1), bias_data,
151153
output_data + bs * w_dims[1]);

0 commit comments

Comments
 (0)