···

       
1
       
1
       
 
       
{ stdenv

     

       
2
       
2
       
+
       
, runCommandCC

     

       
2
       
3
       
 
       
, lib

     

       
3
       
4
       
 
       
, fetchPypi

     

       
4
       
5
       
 
       
, gcc

     

       
5
       
5
       
-
       
, writeScriptBin

     

       
6
       
6
       
 
       
, buildPythonPackage

     

       
7
       
7
       
 
       
, isPyPy

     

       
8
       
8
       
 
       
, pythonOlder

     
···

       
24
       
24
       
 
       
                   && cudnn != null;

     

       
25
       
25
       
 
       


     

       
26
       
26
       
 
       
let

     

       
27
       
27
       
-
       
  extraFlags =

     

       
28
       
28
       
-
       
    lib.optionals cudaSupport [ "-I ${cudatoolkit}/include" "-L ${cudatoolkit}/lib" ]

     

       
29
       
29
       
-
       
    ++ lib.optionals cudnnSupport [ "-I ${cudnn}/include" "-L ${cudnn}/lib" ]

     

       
30
       
30
       
-
       
    ++ lib.optionals cudaSupport [ "-I ${libgpuarray}/include" "-L ${libgpuarray}/lib" ];

     

       
27
       
27
       
+
       
  wrapped = command: buildTop: buildInputs:

     

       
28
       
28
       
+
       
    runCommandCC "${command}-wrapped" { inherit buildInputs; } ''

     

       
29
       
29
       
+
       
      type -P '${command}' || { echo '${command}: not found'; exit 1; }

     

       
30
       
30
       
+
       
      cat > "$out" <<EOF

     

       
31
       
31
       
+
       
      #!$(type -P bash)

     

       
32
       
32
       
+
       
      $(declare -xp | sed -e '/^[^=]\+="\('"''${NIX_STORE//\//\\/}"'\|[^\/]\)/!d')

     

       
33
       
33
       
+
       
      declare -x NIX_BUILD_TOP="${buildTop}"

     

       
34
       
34
       
+
       
      $(type -P '${command}') "\$@"

     

       
35
       
35
       
+
       
      EOF

     

       
36
       
36
       
+
       
      chmod +x "$out"

     

       
37
       
37
       
+
       
    '';

     

       
31
       
38
       
 
       


     

       
32
       
32
       
-
       
  gcc_ = writeScriptBin "g++" ''

     

       
33
       
33
       
-
       
    #!${stdenv.shell}

     

       
34
       
34
       
-
       
    export NIX_CC_WRAPPER_${stdenv.cc.infixSalt}_TARGET_HOST=1

     

       
35
       
35
       
-
       
    export NIX_CFLAGS_COMPILE="$NIX_CFLAGS_COMPILE ${toString extraFlags}"

     

       
36
       
36
       
-
       
    exec ${gcc}/bin/g++ "$@"

     

       
37
       
37
       
-
       
  '';

     

       
39
       
39
       
+
       
  # Theano spews warnings and disabled flags if the compiler isn't named g++

     

       
40
       
40
       
+
       
  cxx_compiler = wrapped "g++" "\\$HOME/.theano"

     

       
41
       
41
       
+
       
    (    stdenv.lib.optional cudaSupport libgpuarray_

     

       
42
       
42
       
+
       
      ++ stdenv.lib.optional cudnnSupport cudnn );

     

       
38
       
43
       
 
       


     

       
39
       
39
       
-
       
  libgpuarray_ = libgpuarray.override { inherit cudaSupport; };

     

       
44
       
44
       
+
       
  libgpuarray_ = libgpuarray.override { inherit cudaSupport cudatoolkit; };

     

       
40
       
45
       
 
       


     

       
41
       
46
       
 
       
in buildPythonPackage rec {

     

       
42
       
47
       
 
       
  pname = "Theano";

     
···

       
50
       
55
       
 
       
  };

     

       
51
       
56
       
 
       


     

       
52
       
57
       
 
       
  postPatch = ''

     

       
53
       
53
       
-
       
    sed -i 's,g++,${gcc_}/bin/g++,g' theano/configdefaults.py

     

       
54
       
54
       
-
       
  '' + lib.optionalString cudnnSupport ''

     

       
55
       
55
       
-
       
    sed -i \

     

       
56
       
56
       
-
       
      -e "s,ctypes.util.find_library('cudnn'),'${cudnn}/lib/libcudnn.so',g" \

     

       
57
       
57
       
-
       
      -e "s/= _dnn_check_compile()/= (True, None)/g" \

     

       
58
       
58
       
-
       
      theano/gpuarray/dnn.py

     

       
58
       
58
       
+
       
    substituteInPlace theano/configdefaults.py \

     

       
59
       
59
       
+
       
      --replace 'StrParam(param, is_valid=warn_cxx)' 'StrParam('\'''${cxx_compiler}'\''', is_valid=warn_cxx)' \

     

       
60
       
60
       
+
       
      --replace 'rc == 0 and config.cxx != ""' 'config.cxx != ""'

     

       
61
       
61
       
+
       
  '' + stdenv.lib.optionalString cudaSupport ''

     

       
62
       
62
       
+
       
    substituteInPlace theano/configdefaults.py \

     

       
63
       
63
       
+
       
      --replace 'StrParam(get_cuda_root)' 'StrParam('\'''${cudatoolkit}'\''')'

     

       
64
       
64
       
+
       
  '' + stdenv.lib.optionalString cudnnSupport ''

     

       
65
       
65
       
+
       
    substituteInPlace theano/configdefaults.py \

     

       
66
       
66
       
+
       
      --replace 'StrParam(default_dnn_base_path)' 'StrParam('\'''${cudnn}'\''')'

     

       
59
       
67
       
 
       
  '';

     

       
60
       
68
       
 
       


     

       
61
       
69
       
 
       
  preCheck = ''