- compilation fix

PaddlePaddle · Jan 28, 2022 · 32018c8 · 32018c8
1 parent 6e8c753
commit 32018c8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/paddle/fluid/operators/elementwise/elementwise_op.h b/paddle/fluid/operators/elementwise/elementwise_op.h
@@ -101,6 +101,7 @@ class ElementwiseOp : public framework::OperatorWithKernel {
       std::vector<int> x_dims_array(max_dim);
       std::vector<int> y_dims_array(max_dim);
       std::vector<int> out_dims_array(max_dim);
+#ifdef PADDLE_WITH_MKLDNN
       // (jczaja): Broadcasting of dims has to be done on Paddle shapes (NHWC)
       // if model is using NHWC.
       bool should_rotate =
@@ -119,15 +120,18 @@ class ElementwiseOp : public framework::OperatorWithKernel {
           y_dims = framework::make_ddim(vdims);
         }
       }
+#endif
 
       GetBroadcastDimsArrays(x_dims, y_dims, x_dims_array.data(),
                              y_dims_array.data(), out_dims_array.data(),
                              max_dim, axis);
+#ifdef PADDLE_WITH_MKLDNN
       // Now rotate shape back if needed (NHWC -> NCHW)
       if (should_rotate) {
         std::rotate(out_dims_array.begin() + 1, out_dims_array.end() - 1,
                     out_dims_array.end());
       }
+#endif
       ctx->SetOutputDim("Out", framework::make_ddim(out_dims_array));
       // to do
       ctx->ShareLoD("X", /*->*/ "Out");