pyrox.dev
pycuda: init at 2016.1
+21
pkgs/development/python-modules/pycuda/compyte.nix
+21
pkgs/development/python-modules/pycuda/compyte.nix
···
1
+
{ mkDerivation
2
+
, fetchFromGitHub
3
+
}:
4
+
5
+
mkDerivation rec {
6
+
name = "compyte-${version}";
7
+
version = "git-20150817";
8
+
9
+
src = fetchFromGitHub {
10
+
owner = "inducer";
11
+
repo = "compyte";
12
+
rev = "ac1c71d46428c14aa1bd1c09d7da19cd0298d5cc";
13
+
sha256 = "1980h017qi52b7fqwm75m481xs2napgdd3fbrzkfc29k085cbign";
14
+
};
15
+
16
+
installPhase = ''
17
+
mkdir -p $out
18
+
cp -r * $out
19
+
'';
20
+
21
+
}
+66
pkgs/development/python-modules/pycuda/default.nix
+66
pkgs/development/python-modules/pycuda/default.nix
···
1
+
{ buildPythonPackage
2
+
, fetchFromGitHub
3
+
, boost
4
+
, numpy
5
+
, pytools
6
+
, pytest
7
+
, decorator
8
+
, appdirs
9
+
, six
10
+
, cudatoolkit
11
+
, python
12
+
, mkDerivation
13
+
, stdenv
14
+
}:
15
+
let
16
+
compyte = import ./compyte.nix {
17
+
inherit mkDerivation fetchFromGitHub;
18
+
};
19
+
in
20
+
buildPythonPackage rec {
21
+
name = "pycuda-${version}";
22
+
version = "2016.1";
23
+
24
+
src = fetchFromGitHub {
25
+
owner = "inducer";
26
+
repo = "pycuda";
27
+
rev = "609817e22c038249f5e9ddd720b3ca5a9d58ca11";
28
+
sha256 = "0kg6ayxsw2gja9rqspy6z8ihacf9jnxr8hzywjwmj1izkv24cff7";
29
+
};
30
+
31
+
preConfigure = ''
32
+
findInputs ${boost} boost_dirs propagated-native-build-inputs
33
+
34
+
export BOOST_INCLUDEDIR=$(echo $boost_dirs | sed -e s/\ /\\n/g - | grep '\-dev')/include
35
+
export BOOST_LIBRARYDIR=$(echo $boost_dirs | sed -e s/\ /\\n/g - | grep -v '\-dev')/lib
36
+
37
+
${python.interpreter} configure.py --boost-inc-dir=$BOOST_INCLUDEDIR \
38
+
--boost-lib-dir=$BOOST_LIBRARYDIR \
39
+
--no-use-shipped-boost \
40
+
--boost-python-libname=boost_python
41
+
'';
42
+
43
+
postInstall = ''
44
+
ln -s ${compyte} $out/${python.sitePackages}/pycuda/compyte
45
+
'';
46
+
47
+
propagatedBuildInputs = [
48
+
numpy
49
+
pytools
50
+
pytest
51
+
decorator
52
+
appdirs
53
+
six
54
+
cudatoolkit
55
+
compyte
56
+
python
57
+
];
58
+
59
+
meta = with stdenv.lib; {
60
+
homepage = https://github.com/inducer/pycuda/;
61
+
description = "CUDA integration for Python.";
62
+
license = licenses.mit;
63
+
maintainers = with maintainers; [ artuuge ];
64
+
};
65
+
66
+
}
+5
pkgs/top-level/python-packages.nix
+5
pkgs/top-level/python-packages.nix
···
7508
7508
};
7509
7509
};
7510
7510
7511
+
pycuda = callPackage ../development/python-modules/pycuda rec {
7512
+
cudatoolkit = pkgs.cudatoolkit75;
7513
+
inherit (pkgs.stdenv) mkDerivation;
7514
+
};
7515
+
7511
7516
python-axolotl = buildPythonPackage rec {
7512
7517
name = "python-axolotl-${version}";
7513
7518
version = "0.1.7";