pyrox.dev
Merge pull request #288738 from samuela/samuela/jaxlib-bin
python3Packages.jaxlib-bin: use `autoAddOpenGLRunpathHook`
authored by Samuel Ainsworth and committed by GitHub 8e573554 ceef74d1
+2
-5
pkgs/development/python-modules/jaxlib/bin.nix
+2
-5
pkgs/development/python-modules/jaxlib/bin.nix
···
14
# * https://github.com/google/jax/issues/5723#issuecomment-913038780
15
16
{ absl-py
17
-
, addOpenGLRunpath
18
, autoPatchelfHook
19
, buildPythonPackage
20
, config
···
33
}:
34
35
let
36
-
inherit (cudaPackagesGoogle) cudatoolkit cudnn cudaVersion;
37
38
version = "0.4.24";
39
···
181
# Prebuilt wheels are dynamically linked against things that nix can't find.
182
# Run `autoPatchelfHook` to automagically fix them.
183
nativeBuildInputs = lib.optionals stdenv.isLinux [ autoPatchelfHook ]
184
-
++ lib.optionals cudaSupport [ addOpenGLRunpath ];
185
# Dynamic link dependencies
186
buildInputs = [ stdenv.cc.cc.lib ];
187
···
196
# * libcuda.so.1 -> opengl driver in /run/opengl-driver/lib
197
preInstallCheck = lib.optional cudaSupport ''
198
shopt -s globstar
199
-
200
-
addOpenGLRunpath $out/**/*.so
201
202
for file in $out/**/*.so; do
203
rpath=$(patchelf --print-rpath $file)
···
14
# * https://github.com/google/jax/issues/5723#issuecomment-913038780
15
16
{ absl-py
17
, autoPatchelfHook
18
, buildPythonPackage
19
, config
···
32
}:
33
34
let
35
+
inherit (cudaPackagesGoogle) autoAddOpenGLRunpathHook cudatoolkit cudnn cudaVersion;
36
37
version = "0.4.24";
38
···
180
# Prebuilt wheels are dynamically linked against things that nix can't find.
181
# Run `autoPatchelfHook` to automagically fix them.
182
nativeBuildInputs = lib.optionals stdenv.isLinux [ autoPatchelfHook ]
183
+
++ lib.optionals cudaSupport [ autoAddOpenGLRunpathHook ];
184
# Dynamic link dependencies
185
buildInputs = [ stdenv.cc.cc.lib ];
186
···
195
# * libcuda.so.1 -> opengl driver in /run/opengl-driver/lib
196
preInstallCheck = lib.optional cudaSupport ''
197
shopt -s globstar
198
199
for file in $out/**/*.so; do
200
rpath=$(patchelf --print-rpath $file)