提交 1a4be55a 编写于 作者: X Xin Pan

Pass cpu build

上级 904fa05f
......@@ -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> {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册