[BUGFIX] Pip Build: use runtime.Features instead of manual check for mkldnn headers (#19195)#19928
Conversation
…mkldnn headers (apache#19195) * use runtime Features instead of manual check * move mxnet package closer to where its used
|
Hey @access2rohit , Thanks for submitting the PR
CI supported jobs: [windows-cpu, unix-gpu, centos-gpu, website, clang, windows-gpu, miscellaneous, sanity, edge, centos-cpu, unix-cpu] Note: |
|
@josephevans @ChaiBapchya Can you review ? |
| libraries.append('CUDA-10.0') | ||
| elif variant.startswith('CU92'): | ||
| libraries.append('CUDA-9.2') | ||
| elif variant.startswith('CU91'): |
There was a problem hiding this comment.
Are we dropping support for all cuda till 9? I guess yes [since cuda11 has launched and we support till latest major version-1]. Can we update the PR title to also reflect the same? Since currently it just talks about mkldnn and this PR changes things more than that.
ChaiBapchya
left a comment
There was a problem hiding this comment.
Minor comment about PR title, rest looks good
Backport #19195 as a part of effort #19911
use runtime Features instead of manual check
move mxnet package closer to where its used
dropping cuda9.1