Pass cpu build

PaddlePaddle · Mar 25, 2018 · 1a4be55 · 1a4be55
1 parent 904fa05
commit 1a4be55
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/paddle/fluid/operators/layer_norm_op.h b/paddle/fluid/operators/layer_norm_op.h
@@ -34,6 +34,7 @@ struct RowwiseMean2D {
                   const framework::Tensor& input, framework::Tensor* vec);
 };
 
+#ifdef PADDLE_WITH_CUDA
 template <typename T>
 class RowwiseMean2D<platform::CUDADeviceContext, T> {
  public:
@@ -55,6 +56,7 @@ class RowwiseMean2D<platform::CUDADeviceContext, T> {
   int right_;
   framework::Tensor divisor_;
 };
+#endif
 
 template <typename T>
 class RowwiseMean2D<platform::CPUDeviceContext, T> {
@@ -78,6 +80,7 @@ struct ColwiseSum2D {
                   const framework::Tensor& input, framework::Tensor* vec);
 };
 
+#ifdef PADDLE_WITH_CUDA
 template <typename T>
 class ColwiseSum2D<platform::CUDADeviceContext, T> {
  public:
@@ -100,6 +103,7 @@ class ColwiseSum2D<platform::CUDADeviceContext, T> {
   int right_;
   framework::Tensor divisor_;
 };
+#endif
 
 template <typename T>
 class ColwiseSum2D<platform::CPUDeviceContext, T> {