646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
100 lines
2.6 KiB
Nix
100 lines
2.6 KiB
Nix
{
|
|
config,
|
|
stdenv,
|
|
lib,
|
|
fetchurl,
|
|
fetchpatch,
|
|
bash,
|
|
cmake,
|
|
opencv4,
|
|
gtest,
|
|
blas,
|
|
gomp,
|
|
llvmPackages,
|
|
perl,
|
|
# mxnet cuda support is turned off, but dependencies like opencv can still be built with cudaSupport
|
|
# and fail to compile without the cudatoolkit
|
|
# mxnet cuda support will not be available, as mxnet requires version <=11
|
|
cudaSupport ? config.cudaSupport,
|
|
cudaPackages ? { },
|
|
}:
|
|
|
|
# mxnet is not maintained, and other projects are migrating away from it.
|
|
# https://github.com/apache/mxnet/issues/21206
|
|
|
|
stdenv.mkDerivation rec {
|
|
pname = "mxnet";
|
|
version = "1.9.1";
|
|
|
|
src = fetchurl {
|
|
name = "apache-mxnet-src-${version}-incubating.tar.gz";
|
|
url = "mirror://apache/incubator/mxnet/${version}/apache-mxnet-src-${version}-incubating.tar.gz";
|
|
hash = "sha256-EephMoF02MKblvNBl34D3rC/Sww3rOZY+T442euMkyI=";
|
|
};
|
|
|
|
patches = [
|
|
# Remove the following two patches when updating mxnet to 2.0.
|
|
(fetchpatch {
|
|
name = "1-auto-disable-sse-for-non-x86.patch";
|
|
url = "https://github.com/apache/incubator-mxnet/commit/55e69871d4cadec51a8bbb6700131065388cb0b9.patch";
|
|
hash = "sha256-uaMpM0F9HRtEBXz2ewB/dlbuKaY5/RineCPUE2T6CHU=";
|
|
})
|
|
(fetchpatch {
|
|
name = "2-auto-disable-sse-for-non-x86.patch";
|
|
url = "https://github.com/apache/incubator-mxnet/commit/c1b96f562f55dfa024ac941d7b104f00e239ee0f.patch";
|
|
excludes = [ "ci/docker/runtime_functions.sh" ];
|
|
hash = "sha256-r1LbC8ueRooW5tTNakAlRSJ+9aR4WXXoEKx895DgOs4=";
|
|
})
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
cmake
|
|
perl
|
|
];
|
|
|
|
buildInputs = [
|
|
opencv4
|
|
gtest
|
|
blas.provider
|
|
]
|
|
++ lib.optional stdenv.cc.isGNU gomp
|
|
++ lib.optional stdenv.cc.isClang llvmPackages.openmp
|
|
++ lib.optionals cudaSupport [
|
|
# needed for OpenCV cmake module
|
|
cudaPackages.cudatoolkit
|
|
];
|
|
|
|
cmakeFlags = [
|
|
(lib.cmakeFeature "CMAKE_POLICY_VERSION_MINIMUM" "3.5")
|
|
"-DUSE_MKL_IF_AVAILABLE=OFF"
|
|
"-DUSE_CUDA=OFF"
|
|
"-DUSE_CUDNN=OFF"
|
|
];
|
|
|
|
env.NIX_CFLAGS_COMPILE = toString [
|
|
# Needed with GCC 12
|
|
"-Wno-error=uninitialized"
|
|
];
|
|
|
|
postPatch = ''
|
|
substituteInPlace 3rdparty/mkldnn/tests/CMakeLists.txt \
|
|
--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
|
|
'';
|
|
|
|
postInstall = ''
|
|
rm "$out"/lib/*.a
|
|
'';
|
|
|
|
meta = with lib; {
|
|
description = "Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler";
|
|
homepage = "https://mxnet.incubator.apache.org/";
|
|
maintainers = [ ];
|
|
license = licenses.asl20;
|
|
platforms = platforms.linux;
|
|
};
|
|
}
|