Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
Log information proposing to use cuDNN when appropriate #4039
What changes were proposed in this pull request?
Logs information to suggest using cuDNN when appropriate and the page where to find more information. The output is displayed only once, but once for each layer...
How was this patch tested?
Running an example with CUDA, but without cuDNN, displays the information, without stopping execution.
AlexDBlack left a comment
The idea is great - but I'm wondering if we use the one-time logger to avoid too many log message? I think currently this will be one per layer, which could be a lot for big nets.
I also might tweak the message slightly, for those who don't yet know what cudnn is (or why they should care) -