Fix CuDNNConvolutionLayer for cuDNN v4
authorFelix Abecassis <fabecassis@nvidia.com>
Thu, 10 Dec 2015 23:11:51 +0000 (15:11 -0800)
committerFelix Abecassis <fabecassis@nvidia.com>
Thu, 10 Dec 2015 23:11:51 +0000 (15:11 -0800)
commiteb2b848df173f7a07eb0d76a432c5d4badca7ba6
treeabbb23064fcd7975938dd8cfd70575c4a95d3963
parent541386054c5e0664458b1a7fc643a3ddc7f6c39e
Fix CuDNNConvolutionLayer for cuDNN v4

Add a macro to check the current cuDNN version
include/caffe/util/cudnn.hpp
src/caffe/layers/cudnn_conv_layer.cu