blob: 616e55f3704d5b23df8aa84b37ce0846216ad4ac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
{ pkgs ? import <nixpkgs> { }, ... }:
pkgs.stdenvNoCC.mkDerivation rec {
name = "pycl-env";
env = pkgs.buildEnv { name = name; paths = buildInputs; };
buildInputs = let
boltzgen = pkgs.python3.pkgs.buildPythonPackage rec {
pname = "boltzgen";
version = "0.1";
#src = pkgs.fetchFromGitHub {
#owner = "KnairdA";
#repo = "boltzgen";
#rev = "v0.1";
#sha256 = "072kx4jrzd0g9rn63hjb0yic7qhbga47lp2vbz7rq3gvkqv1hz4d";
#};
src = ~/projects/dev/boltzgen;
propagatedBuildInputs = with pkgs.python37Packages; [
sympy
numpy
Mako
];
doCheck = false;
};
local-python = pkgs.python3.withPackages (python-packages: with python-packages; [
boltzgen
numpy
pyopencl setuptools
matplotlib
]);
in [
local-python
pkgs.opencl-info
];
shellHook = ''
export NIX_SHELL_NAME="${name}"
export PYOPENCL_COMPILER_OUTPUT=1
export PYTHONPATH="$PWD:$PYTHONPATH"
'';
}
|