-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
Hey @mk-61 , Thanks for submitting the PR
CI supported jobs: [website, sanity, unix-cpu, windows-cpu, edge, windows-gpu, centos-cpu, centos-gpu, unix-gpu, clang, miscellaneous] Note: |
@mxnet-bot run ci [miscellaneous, unix-gpu] |
Jenkins CI successfully triggered : [unix-gpu, miscellaneous] |
@mxnet-bot run ci [unix-gpu] |
Jenkins CI successfully triggered : [unix-gpu] |
@mxnet-bot run ci [unix-gpu] |
Jenkins CI successfully triggered : [unix-gpu] |
Thanks again for your responsiveness to our review comments, and patience throughout these recent CI issues (e.g. #20738). The problem with the apt repo server seems to have been solved, and this PR has passed without new work-arounds added. LGTM. |
Description
This change ports Convolution and Deconvolution operations to cuDNN v8 API.
Legacy API support is dropped, as per this RFC: #20618.
The change also includes some cuDNN v8 API general support stuff, to be re-used later when more operations are ported to the v8 API.
Finally, auto-tuning functionality is moved from cuDNN into MXNet, hence some memory management changes were required.
Checklist
Essentials