We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8f9a8c4 commit 9b64aacCopy full SHA for 9b64aac
paddle/fluid/operators/math/pooling.h
@@ -82,7 +82,7 @@ class AvgPoolGrad {
82
* This is different from average pooling. So we rewrite the max_pool_grad:
83
* MaxPool2dGradFunctor, MaxPool3dGradFunctor.
84
*/
85
-
+#ifdef PADDLE_WITH_CUDA
86
template <typename PoolProcess, typename T>
87
class Pool2dDirectCUDAFunctor {
88
public:
@@ -93,6 +93,7 @@ class Pool2dDirectCUDAFunctor {
93
const std::vector<int>& paddings, PoolProcess pool_compute,
94
bool exclusive, T* output, cudaStream_t stream);
95
};
96
+#endif
97
98
template <typename DeviceContext, typename PoolProcess, typename T>
99
class Pool2dFunctor {
0 commit comments