-
Notifications
You must be signed in to change notification settings - Fork 18.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
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