colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mxnet: patch to fix build on aarch64-darwin

Browse Source 
The patches autodetect when the system is x86 and supports SSE.
This commit is contained in:
Theodore Ni 2022-09-22 16:50:21 -07:00
parent c13952175c
commit 5ac4f45f72
No known key found for this signature in database
GPG Key ID: 48B67583BDDD4474
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pkgs/applications/science/math/mxnet/default.nix
View File

@ -1,4 +1,4 @@
{ config, stdenv, lib, fetchurl, bash, cmake
{ config, stdenv, lib, fetchurl, fetchpatch, bash, cmake
, opencv3, gtest, blas, gomp, llvmPackages, perl
, cudaSupport ? config.cudaSupport or false, cudaPackages ? {}, nvidia_x11
, cudnnSupport ? cudaSupport
@ -21,6 +21,21 @@ stdenv.mkDerivation rec {
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 = [ opencv3 gtest blas.provider ]