From 20ad7ebd0be3f79b7e795fd6fac71b1a808692cf Mon Sep 17 00:00:00 2001
From: lltcggie <lltcggie@gmail.com>
Date: Sat, 1 Dec 2018 14:45:38 +0900
Subject: [PATCH] =?UTF-8?q?Pooling=E3=81=8C=E5=BC=B7=E5=88=B6=E7=9A=84?=
 =?UTF-8?q?=E3=81=ABcuDNN=E3=82=92=E4=BD=BF=E3=81=A3=E3=81=A6=E3=81=97?=
 =?UTF-8?q?=E3=81=BE=E3=81=86=E3=81=AE=E3=82=92=E4=BF=AE=E6=AD=A3=20resolv?=
 =?UTF-8?q?ed=20#133?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 common/cNet.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/common/cNet.cpp b/common/cNet.cpp
index 86d65b2..5fdea92 100644
--- a/common/cNet.cpp
+++ b/common/cNet.cpp
@@ -423,6 +423,13 @@ Waifu2x::eWaifu2xError cNet::SetParameter(caffe::NetParameter &param, const std:
 			else
 				layer_param->mutable_sigmoid_param()->set_engine(caffe::SigmoidParameter_Engine_CAFFE);
 		}
+		else if (type == "Pooling")
+		{
+			if (process == "cudnn")
+				layer_param->mutable_pooling_param()->set_engine(caffe::PoolingParameter_Engine_CUDNN);
+			else
+				layer_param->mutable_pooling_param()->set_engine(caffe::PoolingParameter_Engine_CAFFE);
+		}
 	}
 
 	return Waifu2x::eWaifu2xError_OK;