mxnet: fix library dependencies.
- mxnet now builds the open source MKL DNN library. However, this was not included with the build. - mxnet builds its own OpenMP library, even when OpenMP is provided by the system. This leads to problems, so remove the included OpenMP implementation: https://github.com/apache/incubator-mxnet/pull/12160
This commit is contained in:
parent
da0c385a69
commit
590ce7d1ae
@ -34,11 +34,14 @@ stdenv.mkDerivation rec {
|
|||||||
postPatch = ''
|
postPatch = ''
|
||||||
substituteInPlace 3rdparty/mkldnn/tests/CMakeLists.txt \
|
substituteInPlace 3rdparty/mkldnn/tests/CMakeLists.txt \
|
||||||
--replace "/bin/bash" "${bash}/bin/bash"
|
--replace "/bin/bash" "${bash}/bin/bash"
|
||||||
|
|
||||||
|
# Build against the system version of OpenMP.
|
||||||
|
# https://github.com/apache/incubator-mxnet/pull/12160
|
||||||
|
rm -rf 3rdparty/openmp
|
||||||
'';
|
'';
|
||||||
|
|
||||||
installPhase = ''
|
postInstall = ''
|
||||||
install -Dm755 libmxnet.so $out/lib/libmxnet.so
|
rm "$out"/lib/*.a
|
||||||
cp -r ../include $out
|
|
||||||
'';
|
'';
|
||||||
|
|
||||||
enableParallelBuilding = true;
|
enableParallelBuilding = true;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user