python3Packages.torch: switch to apple-sdk_13

Switch to apple-sdk_13 on Darwin. This enables the use of MPS (Metal Performance Shaders) on macOS. Signed-off-by: Mika Tammi <[email protected]>
NixOS · Oct 27, 2024 · 58971c7 · 58971c7
1 parent cae0fd4
commit 58971c7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix
@@ -35,7 +35,7 @@
   removeReferencesTo,
 
   # Build inputs
-  darwin,
+  apple-sdk_13,
   numactl,
 
   # dependencies
@@ -519,9 +519,7 @@ buildPythonPackage rec {
     ++ lib.optionals (cudaSupport || rocmSupport) [ effectiveMagma ]
     ++ lib.optionals stdenv.hostPlatform.isLinux [ numactl ]
     ++ lib.optionals stdenv.hostPlatform.isDarwin [
-      darwin.apple_sdk.frameworks.Accelerate
-      darwin.apple_sdk.frameworks.CoreServices
-      darwin.libobjc
+      apple-sdk_13
     ]
     ++ lib.optionals tritonSupport [ _tritonEffective ]
     ++ lib.optionals MPISupport [ mpi ]