From 7ba3d1e4bdcbb98d152bba176fe6edbddb080b38 Mon Sep 17 00:00:00 2001
From: chengduoZH <zhaochengduo@163.com>
Date: Mon, 13 Nov 2017 17:32:50 +0800
Subject: [PATCH] add cudnn_pool3d_op

---
 paddle/operators/pool_cudnn_op.cc | 8 ++++++++
 paddle/operators/pool_cudnn_op.cu | 3 +++
 2 files changed, 11 insertions(+)

diff --git a/paddle/operators/pool_cudnn_op.cc b/paddle/operators/pool_cudnn_op.cc
index f962d9e3e..06cf1c0d2 100644
--- a/paddle/operators/pool_cudnn_op.cc
+++ b/paddle/operators/pool_cudnn_op.cc
@@ -23,3 +23,11 @@ REGISTER_OP_CPU_KERNEL(pool2d_cudnn,
                        ops::PoolKernel<paddle::platform::CPUPlace, float>);
 REGISTER_OP_CPU_KERNEL(pool2d_cudnn_grad,
                        ops::PoolGradKernel<paddle::platform::CPUPlace, float>)
+
+REGISTER_OP(pool3d_cudnn, ops::PoolOp, ops::Pool3dOpMaker, pool3d_cudnn_grad,
+            ops::PoolOpGrad);
+
+REGISTER_OP_CPU_KERNEL(pool3d_cudnn,
+                       ops::PoolKernel<paddle::platform::CPUPlace, float>);
+REGISTER_OP_CPU_KERNEL(pool3d_cudnn_grad,
+                       ops::PoolGradKernel<paddle::platform::CPUPlace, float>)
diff --git a/paddle/operators/pool_cudnn_op.cu b/paddle/operators/pool_cudnn_op.cu
index 8711567b9..ccfe35def 100644
--- a/paddle/operators/pool_cudnn_op.cu
+++ b/paddle/operators/pool_cudnn_op.cu
@@ -153,3 +153,6 @@ namespace ops = paddle::operators;
 
 REGISTER_OP_GPU_KERNEL(pool2d_cudnn, ops::PoolCudnnOpKernel<float>);
 REGISTER_OP_GPU_KERNEL(pool2d_cudnn_grad, ops::PoolCudnnGradOpKernel<float>);
+
+REGISTER_OP_GPU_KERNEL(pool3d_cudnn, ops::PoolCudnnOpKernel<float>);
+REGISTER_OP_GPU_KERNEL(pool3d_cudnn_grad, ops::PoolCudnnGradOpKernel<float>);
\ No newline at end of file
-- 
GitLab