From 1ac0ad8f0d2043a43e6c78b6144c953ae34c841d Mon Sep 17 00:00:00 2001
From: Md Amir Sohail <mdas442006@gmail.com>
Date: Sun, 29 Jul 2018 21:26:45 +0530
Subject: [PATCH] For CuDNN higher version support

Check for setConvolutionDesc
---
 include/caffe/util/cudnn.hpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/include/caffe/util/cudnn.hpp b/include/caffe/util/cudnn.hpp
index a7d8dbb..d9cb435 100644
--- a/include/caffe/util/cudnn.hpp
+++ b/include/caffe/util/cudnn.hpp
@@ -109,8 +109,15 @@ template <typename Dtype>
 inline void setConvolutionDesc(cudnnConvolutionDescriptor_t* conv,
     cudnnTensorDescriptor_t bottom, cudnnFilterDescriptor_t filter,
     int pad_h, int pad_w, int stride_h, int stride_w) {
+#if CUDNN_VERSION_MIN(6, 0, 0) 					//Added from here (amir)
+    // For CuDNN higher version
   CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
+      pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION,
+      dataType<Dtype>::type));
+#else
+    CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
       pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION));
+#endif 								//Added upto this (amir)
 }
 
 template <typename Dtype>