diff options
author | Ryan Burns <rtburns@protonmail.com> | 2021-09-23 01:40:27 -0700 |
---|---|---|
committer | Ryan Burns <rtburns@protonmail.com> | 2021-09-23 02:16:17 -0700 |
commit | 813c400edc0802a16c9c4b305f111b7f9b22c537 (patch) | |
tree | 24dbd3bffc7707e8df4e668d306c019f526fc194 /pkgs/applications/science/math/cntk | |
parent | 0c350944709004cf29645601d6e886dcaeba6a0c (diff) | |
download | nixpkgs-813c400edc0802a16c9c4b305f111b7f9b22c537.tar nixpkgs-813c400edc0802a16c9c4b305f111b7f9b22c537.tar.gz nixpkgs-813c400edc0802a16c9c4b305f111b7f9b22c537.tar.bz2 nixpkgs-813c400edc0802a16c9c4b305f111b7f9b22c537.tar.lz nixpkgs-813c400edc0802a16c9c4b305f111b7f9b22c537.tar.xz nixpkgs-813c400edc0802a16c9c4b305f111b7f9b22c537.tar.zst nixpkgs-813c400edc0802a16c9c4b305f111b7f9b22c537.zip |
cntk: fix build with protobuf 3.18+
Diffstat (limited to 'pkgs/applications/science/math/cntk')
-rw-r--r-- | pkgs/applications/science/math/cntk/default.nix | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/pkgs/applications/science/math/cntk/default.nix b/pkgs/applications/science/math/cntk/default.nix index c007490c94d..ba2225b903f 100644 --- a/pkgs/applications/science/math/cntk/default.nix +++ b/pkgs/applications/science/math/cntk/default.nix @@ -1,4 +1,5 @@ { lib, stdenv, fetchgit, fetchFromGitHub, cmake +, fetchpatch , openblas, blas, lapack, opencv3, libzip, boost, protobuf, mpi , onebitSGDSupport ? false , cudaSupport ? false, addOpenGLRunpath, cudatoolkit, nvidia_x11 @@ -28,6 +29,26 @@ in stdenv.mkDerivation rec { sha256 = "18l9k7s966a26ywcf7flqyhm61788pcb9fj3wk61jrmgkhy2pcns"; }; + patches = [ + # Fix build with protobuf 3.18+ + # Remove with onnx submodule bump to 1.9+ + (fetchpatch { + url = "https://github.com/onnx/onnx/commit/d3bc82770474761571f950347560d62a35d519d7.patch"; + extraPrefix = "Source/CNTKv2LibraryDll/proto/onnx/onnx_repo/"; + stripLen = 1; + sha256 = "00raqj8wx30b06ky6cdp5vvc1mrzs7hglyi6h58hchw5lhrwkzxp"; + }) + ]; + + postPatch = '' + # Fix build with protobuf 3.18+ + substituteInPlace Source/CNTKv2LibraryDll/Serialization.cpp \ + --replace 'SetTotalBytesLimit(INT_MAX, INT_MAX)' \ + 'SetTotalBytesLimit(INT_MAX)' \ + --replace 'SetTotalBytesLimit(limit, limit)' \ + 'SetTotalBytesLimit(limit)' + ''; + nativeBuildInputs = [ cmake ] ++ lib.optional cudaSupport addOpenGLRunpath; # Force OpenMPI to use g++ in PATH. |