Unverified Commit 61220d76 authored by Mika Tammi's avatar Mika Tammi
Browse files

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: default avatarMika Tammi <mikatammi@gmail.com>
parent 1610bc14

pkgs/development/python-modules/torch/default.nix

+5 −4
Original line number Diff line number Diff line
@@ -35,7 +35,7 @@ 
  removeReferencesTo,



  # Build inputs

  darwin,

  apple-sdk_13,

  numactl,



  # dependencies
@@ -364,6 +364,9 @@ buildPythonPackage rec { 
  # NB technical debt: building without NNPACK as workaround for missing `six`

  USE_NNPACK = 0;



  # Explicitly enable MPS for Darwin

  USE_MPS = setBool stdenv.hostPlatform.isDarwin;



  cmakeFlags =

    [

      # (lib.cmakeBool "CMAKE_FIND_DEBUG_MODE" true)
@@ -519,9 +522,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 ]