b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

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

Browse source 
python310Packages.torch: restrict platforms
This commit is contained in:
Weijia Wang 2023-05-17 21:53:41 +03:00 committed by GitHub
parent 2515b0e0dc a8a453caf8
commit d1bc3181fd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pkgs/development/python-modules/torch/default.nix
View file

@ -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
};
}