Merge pull request #1875 from JAEarly/master

overrides: add setup for torchaudio
nix-community · Jan 7, 2025 · 4d260d9 · 4d260d9
2 parents 5c4de98 + 9f28fb3
commit 4d260d9
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/overrides/default.nix b/overrides/default.nix
@@ -3966,6 +3966,19 @@ lib.composeManyExtensions [
         ];
       });
 
+      torchaudio = prev.torchaudio.overridePythonAttrs (old: {
+        autoPatchelfIgnoreMissingDeps = true;
+
+        # (no patchelf on darwin, since no elves there.)
+        preFixup = lib.optionals (!stdenv.isDarwin) ''
+          addAutoPatchelfSearchPath "${final.torch}/${final.python.sitePackages}/torch/lib"
+        '';
+
+        buildInputs = old.buildInputs or [ ] ++ [
+          final.torch
+        ];
+      });
+
       torchvision = prev.torchvision.overridePythonAttrs (old: {
         autoPatchelfIgnoreMissingDeps = true;