-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[BUGFIX] Ubuntu 20.04: support for in-distro Intel MKL libraries #19766
Conversation
Hey @akarbown , Thanks for submitting the PR
CI supported jobs: [clang, centos-cpu, unix-gpu, centos-gpu, windows-cpu, unix-cpu, miscellaneous, website, sanity, windows-gpu, edge] Note: |
734ccf8
to
f737c59
Compare
@mxnet-bot run ci [unix-cpu] |
Jenkins CI successfully triggered : [unix-cpu] |
f737c59
to
c09039c
Compare
@akarbown the CI fails because the tensorrt gpu build uses Ubuntu 18.04. The reason is that TensorRT was not available for 20.04 yet (see https://gitlab.com/nvidia/container-images/cuda/-/issues/99) But for 18.04 intel-mkl is not available, so you may need to add a bash if/else statement to install
Another option is to wait until nvidia makes tensorrt available on 20.04 |
I thought that it would be able to remove apt repo.
ok, thanks!
but it seems that there is no sense to check it, isn't it? |
Yes, until https://gitlab.com/nvidia/container-images/cuda/-/issues/99 it won't work. cc @TristonC as missing TensorRT for Ubuntu 20.04 is causing issues in this PR. |
402a2b1
to
a182d89
Compare
@mxnet-bot run ci [sanity] |
Jenkins CI successfully triggered : [sanity] |
f9bd9a3
to
b37e1d3
Compare
@mxnet-bot run ci [unix-gpu] |
Jenkins CI successfully triggered : [unix-gpu] |
@mxnet-bot run ci [unix-gpu] |
Jenkins CI successfully triggered : [unix-gpu] |
@mxnet-bot run ci [miscellaneous] |
Jenkins CI successfully triggered : [miscellaneous] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, thank you!
b37e1d3
to
1cd64b8
Compare
@mxnet-bot run ci [centos-gpu, unix-cpu, unix-gpu, website, windows-gpu, windows-cpu] |
Jenkins CI successfully triggered : [windows-cpu, unix-cpu, windows-gpu, centos-gpu, website, unix-gpu] |
Ubuntu 20.04: - add support for intel-mkl distribution version; - use FindBLAS cmake function to look for the intel-mkl; - add FindBLAS.cmake file upstream version (due to Intel10_64_dyn & --start-group/--end-gropup options support) - remove FindMKL.cmake; - preserve Ubuntu 18.04 images for TensorRT pipeline; - preserve MKL libraries linking support on non-Ubuntu 20.04 OSes (w/o FindMKL.cmake file).
1cd64b8
to
39b271a
Compare
@akarbown as you marked this WIP, can you clarify what more you like to include? |
I just wanted to make sure that after I squashed the commits into one all the tests passes without issues. |
Thank you. Actually we can squash the commits automatically during "merge". I think the tests passed previously, but now there seems to be an issue with np.linspace? I think it's due to the new numpy release and to workaround the issue you can force numpy<1.20.0 |
@mxnet-bot run ci [unix-gpu] |
Jenkins CI successfully triggered : [unix-gpu] |
Description
Since Ubuntu 20.04 Intel MKL packages are available in the distribution. Thus, its search could be simplified by using the unified cmake FindBLAS function.
Changes
--start-group/--end-gropup options support)
(w/o FindMKL.cmake file).
Checklist
Essentials
Comments
Reconsidering MKL_USE_SINGLE_DYNAMIC_LIBRARY option as the potential solution for 18255. This needs more investigation.