提交 9b64aac4 编写于 作者: N nhzlx

add macro for pool2dDirectCUDAFunctor

test=develop
上级 8f9a8c45
...@@ -82,7 +82,7 @@ class AvgPoolGrad { ...@@ -82,7 +82,7 @@ class AvgPoolGrad {
* This is different from average pooling. So we rewrite the max_pool_grad: * This is different from average pooling. So we rewrite the max_pool_grad:
* MaxPool2dGradFunctor, MaxPool3dGradFunctor. * MaxPool2dGradFunctor, MaxPool3dGradFunctor.
*/ */
#ifdef PADDLE_WITH_CUDA
template <typename PoolProcess, typename T> template <typename PoolProcess, typename T>
class Pool2dDirectCUDAFunctor { class Pool2dDirectCUDAFunctor {
public: public:
...@@ -93,6 +93,7 @@ class Pool2dDirectCUDAFunctor { ...@@ -93,6 +93,7 @@ class Pool2dDirectCUDAFunctor {
const std::vector<int>& paddings, PoolProcess pool_compute, const std::vector<int>& paddings, PoolProcess pool_compute,
bool exclusive, T* output, cudaStream_t stream); bool exclusive, T* output, cudaStream_t stream);
}; };
#endif
template <typename DeviceContext, typename PoolProcess, typename T> template <typename DeviceContext, typename PoolProcess, typename T>
class Pool2dFunctor { class Pool2dFunctor {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册