Fix cuDNN RNN dtype_with_fallback_ bug#16671
Conversation
DickJC123
left a comment
There was a problem hiding this comment.
Please replace CUDNN_VERSION with cudnnGetVersion(). Since we only warn users if the linked-against version mismatches the compiled-against version, I think we need to consider that case here. In most cases, the user would just not receive the additional cuDNN functionality. However for the issue this PR is addressing, a user that compiled against cuDNN v7.4, but then ran with cuDNN v7.5 would subtly have their RNNs converted from pseudo-fp16 to true-fp16. Use of cudnnGetVersion() corrects this and delivers consistent pseudo-fp16 behavior. Sorry I didn't think of this when creating the original issue.
32c713f to
2b95500
Compare
|
I've selectively retriggered the CI, as I don't think the recent CI failures are related to this PR. The Clojure CPU Integration error was a bit strange, so I've filed it as issue #16674 for someone to look at. |
|
Thanks @DickJC123 |
Description
#16670
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Changes
Comments
@DickJC123