blob: b91c77277eee0a591ea51820fa1ea5ac887f4d05 (
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
{ pkgs ? import <nixpkgs> { }, ... }:
pkgs.stdenvNoCC.mkDerivation rec {
name = "boltzgen-env";
env = pkgs.buildEnv { name = name; paths = buildInputs; };
buildInputs = let
custom-python = let
packageOverrides = self: super: {
pyopencl = super.pyopencl.overridePythonAttrs(old: rec {
buildInputs = with pkgs; [
opencl-headers ocl-icd python37Packages.pybind11
libGLU_combined
];
# Enable OpenGL integration and fix build
preBuild = ''
python configure.py --cl-enable-gl
export HOME=/tmp/pyopencl
'';
});
};
in pkgs.python3.override { inherit packageOverrides; };
pyevtk = pkgs.python3.pkgs.buildPythonPackage rec {
pname = "PyEVTK";
version = "1.2.1";
src = pkgs.fetchFromGitHub {
owner = "paulo-herrera";
repo = "PyEVTK";
rev = "v1.2.1";
sha256 = "1p2459dqvgakywvy5d31818hix4kic6ks9j4m582ypxyk5wj1ksz";
};
buildInputs = with pkgs.python37Packages; [
numpy
];
doCheck = false;
};
local-python = custom-python.withPackages (python-packages: with python-packages; [
numpy
sympy
pyopencl
pyopengl
pyrr
Mako
pyevtk
]);
in [
local-python
pkgs.opencl-info
];
shellHook = ''
export NIX_SHELL_NAME="${name}"
export PYOPENCL_COMPILER_OUTPUT=1
export PYTHONPATH="$PWD/boltzgen:$PYTHONPATH"
'';
}
|