提交 742300ba 编写于 作者: T tensor-tang

fix unkown omp pragmas

上级 b9dbb7c5
...@@ -145,7 +145,9 @@ class FCOpKernel : public framework::OpKernel<T> { ...@@ -145,7 +145,9 @@ class FCOpKernel : public framework::OpKernel<T> {
if (bias) { if (bias) {
const T* bias_data = bias->data<T>(); const T* bias_data = bias->data<T>();
#ifdef PADDLE_WITH_MKLML
#pragma omp parallel for if (FLAGS_paddle_num_threads > 1) #pragma omp parallel for if (FLAGS_paddle_num_threads > 1)
#endif
for (int bs = 0; bs < in_dims[0]; bs++) { for (int bs = 0; bs < in_dims[0]; bs++) {
blas.AXPY(w_dims[1], static_cast<T>(1), bias_data, blas.AXPY(w_dims[1], static_cast<T>(1), bias_data,
output_data + bs * w_dims[1]); output_data + bs * w_dims[1]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册