diff options
| author | Felix Abecassis <fabecassis@nvidia.com> | 2015-12-10 15:11:51 -0800 |
|---|---|---|
| committer | Felix Abecassis <fabecassis@nvidia.com> | 2015-12-10 15:11:51 -0800 |
| commit | eb2b848df173f7a07eb0d76a432c5d4badca7ba6 (patch) | |
| tree | abbb23064fcd7975938dd8cfd70575c4a95d3963 /include | |
| parent | 541386054c5e0664458b1a7fc643a3ddc7f6c39e (diff) | |
| download | caffeonacl-eb2b848df173f7a07eb0d76a432c5d4badca7ba6.tar.gz caffeonacl-eb2b848df173f7a07eb0d76a432c5d4badca7ba6.tar.bz2 caffeonacl-eb2b848df173f7a07eb0d76a432c5d4badca7ba6.zip | |
Fix CuDNNConvolutionLayer for cuDNN v4
Add a macro to check the current cuDNN version
Diffstat (limited to 'include')
| -rw-r--r-- | include/caffe/util/cudnn.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/include/caffe/util/cudnn.hpp b/include/caffe/util/cudnn.hpp index b531dd5f..8a7e17c6 100644 --- a/include/caffe/util/cudnn.hpp +++ b/include/caffe/util/cudnn.hpp @@ -7,6 +7,9 @@ #include "caffe/common.hpp" #include "caffe/proto/caffe.pb.h" +#define CUDNN_VERSION_MIN(major, minor, patch) \ + (CUDNN_VERSION >= (major * 1000 + minor * 100 + patch)) + #define CUDNN_CHECK(condition) \ do { \ cudnnStatus_t status = condition; \ |