Unverified Commit d1bc3181 authored by Weijia Wang's avatar Weijia Wang Committed by GitHub
Browse files

Merge pull request #232457 from wegank/cudatoolkit-eval-1 

python310Packages.torch: restrict platforms
parents 2515b0e0 a8a453ca

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

+1 −1
Original line number Diff line number Diff line
@@ -414,7 +414,7 @@ in buildPythonPackage rec { 
    homepage = "https://pytorch.org/";

    license = licenses.bsd3;

    maintainers = with maintainers; [ teh thoughtpolice tscholak ]; # tscholak esp. for darwin-related builds

    platforms = with platforms; linux ++ lib.optionals (!cudaSupport || !rocmSupport) darwin;

    platforms = with platforms; linux ++ lib.optionals (!cudaSupport && !rocmSupport) darwin;

    broken = rocmSupport && cudaSupport; # CUDA and ROCm are mutually exclusive

  };

}