python3Packages.torch*: use pkgs.autoAddDriverRunpath

2024-03-31 10:20:37 -07:00
parent 65c4c21a2b
commit 7cd68c3dc8
2 changed files with 3 additions and 1 deletions
@@ -1,5 +1,6 @@
 { lib, stdenv
 , buildPythonPackage
+, autoAddDriverRunpath
 , fetchurl
 , python
 , pythonAtLeast
@@ -40,7 +41,7 @@ in buildPythonPackage {
  nativeBuildInputs = lib.optionals stdenv.isLinux [
    addOpenGLRunpath
    autoPatchelfHook
-    cudaPackages.autoAddDriverRunpath
+    autoAddDriverRunpath
  ];

  buildInputs = lib.optionals stdenv.isLinux (with cudaPackages; [
@@ -1,5 +1,6 @@
 { stdenv, lib, fetchFromGitHub, fetchpatch, buildPythonPackage, python,
  config, cudaSupport ? config.cudaSupport, cudaPackages,
+  autoAddDriverRunpath,
  effectiveMagma ?
  if cudaSupport then magma-cuda-static
  else if rocmSupport then magma-hip