Fix CuDNNConvolutionLayer for cuDNN v4#3439
Conversation
|
The CMake website is under maintenance, causing the automated cmake builds to fail. |
Add a macro to check the current cuDNN version
|
@flx42 will there be further ifdef guards for v4? If not this is simple and done. I'm happy to keep v3 compatibility and add v4 if it doesn't obscure the code. So far we have followed the current cuDNN version and dropped old versions (because of API changes). Thanks for the PR. |
|
This is the only change until we get to more feature support like batch normalization. We should generally support the macro to better deal with API change-overs. |
|
For v3 and v4 only, I believe you could simply replace |
|
Ok, agreed on the macro and this is fine as-is. |
Add ifdef in CuDNNConvolutionLayer for cuDNN v4
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Latest CUDA versions are all compatible, and Caffe has been compatible with cuDNN v4 since PR BVLC#3439
Add a macro to check the current cuDNN version