Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 1a4be55a 编写于 作者: X Xin Pan
浏览文件
操作

Pass cpu build

上级 904fa05f
隐藏空白更改
内联 并排
Showing with 4 addition and 0 deletion
paddle/fluid/operators/layer_norm_op.h
浏览文件 @ 1a4be55a
...@@ -34,6 +34,7 @@ struct RowwiseMean2D { ...@@ -34,6 +34,7 @@ struct RowwiseMean2D {
const framework::Tensor& input, framework::Tensor* vec); const framework::Tensor& input, framework::Tensor* vec);
}; };
#ifdef PADDLE_WITH_CUDA
template <typename T> template <typename T>
class RowwiseMean2D<platform::CUDADeviceContext, T> { class RowwiseMean2D<platform::CUDADeviceContext, T> {
public: public:
...@@ -55,6 +56,7 @@ class RowwiseMean2D<platform::CUDADeviceContext, T> { ...@@ -55,6 +56,7 @@ class RowwiseMean2D<platform::CUDADeviceContext, T> {
int right_; int right_;
framework::Tensor divisor_; framework::Tensor divisor_;
}; };
#endif
template <typename T> template <typename T>
class RowwiseMean2D<platform::CPUDeviceContext, T> { class RowwiseMean2D<platform::CPUDeviceContext, T> {
...@@ -78,6 +80,7 @@ struct ColwiseSum2D { ...@@ -78,6 +80,7 @@ struct ColwiseSum2D {
const framework::Tensor& input, framework::Tensor* vec); const framework::Tensor& input, framework::Tensor* vec);
}; };
#ifdef PADDLE_WITH_CUDA
template <typename T> template <typename T>
class ColwiseSum2D<platform::CUDADeviceContext, T> { class ColwiseSum2D<platform::CUDADeviceContext, T> {
public: public:
...@@ -100,6 +103,7 @@ class ColwiseSum2D<platform::CUDADeviceContext, T> { ...@@ -100,6 +103,7 @@ class ColwiseSum2D<platform::CUDADeviceContext, T> {
int right_; int right_;
framework::Tensor divisor_; framework::Tensor divisor_;
}; };
#endif
template <typename T> template <typename T>
class ColwiseSum2D<platform::CPUDeviceContext, T> { class ColwiseSum2D<platform::CPUDeviceContext, T> {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或