python3Packages.jaxlib: use pkgs.autoAddDriverRunpath

2024-03-31 07:46:39 -07:00
parent c8f981bfaa
commit 926ec0e1eb
1 changed files with 2 additions and 1 deletions
@@ -4,6 +4,7 @@

  # Build-time dependencies:
 , addOpenGLRunpath
+, autoAddDriverRunpath
 , bazel_6
 , binutils
 , buildBazelPackage
@@ -51,7 +52,7 @@
 }@inputs:

 let
-  inherit (cudaPackagesGoogle) autoAddDriverRunpath cudaFlags cudaVersion cudnn nccl;
+  inherit (cudaPackagesGoogle) cudaFlags cudaVersion cudnn nccl;

  pname = "jaxlib";
  version = "0.4.24";