Commit a8a453ca authored by Weijia Wang's avatar Weijia Wang
Browse files

python310Packages.torch: restrict platforms

parent 84c6233a

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

  };

}